Modification d'un sous-ensemble de lignes dans un dataframe pandas

Question

Supposons que j'ai un DataFrame pandas avec deux colonnes, A et B. J'aimerais modifier ce DataFrame (ou en créer une copie) pour que B soit toujours NaN lorsque A est égal à 0. Comment puis-je y parvenir ?

J'ai essayé ce qui suit

df['A'==0]['B'] = np.nan

et

df['A'==0]['B'].values.fill(np.nan)

sans succès.