Wie setzen Werte im Pandas-Datenrahmen auf NaN-Werten einer anderen Spalte?

Question

Ich habe Datenframe namens df mit Originalform (4361, 15). Einige Werte der Spalte agefm sind NaN. Sehen Sie nur:

> df[df.agefm.isnull() == True].agefm.shape 
(2282,) 

Dann erstelle ich neue Spalte und stellen Sie alle Werte auf 0:

df['nevermarr'] = 0 

So würde Ich mag nevermarr Wert auf 1 setzen, dann in dieser Zeile agefm ist Nan:

df[df.agefm.isnull() == True].nevermarr = 1 

nichts geändert:

> df['nevermarr'].sum() 
0 

Was mache ich falsch?

Answer 1

Die beste Verwendung ist numpy.where:

df['nevermarr'] = np.where(df.agefm.isnull(), 1, 0) 
print (df) 
    agefm nevermarr 
0 NaN   1 
1 5.0   0 
2 6.0   0 

Oder verwenden loc kann ==True weggelassen werden:

df.loc[df.agefm.isnull(), 'nevermarr'] = 1 

Oder mask:

df['nevermarr'] = df.nevermarr.mask(df.agefm.isnull(), 1) 
print (df) 
    agefm nevermarr 
0 NaN   1 
1 5.0   2 
2 6.0   3 

Probe:

import pandas as pd 
import numpy as np 

df = pd.DataFrame({'nevermarr':[7,2,3], 
        'agefm':[np.nan,5,6]}) 

print (df) 
    agefm nevermarr 
0 NaN   7 
1 5.0   2 
2 6.0   3 

df.loc[df.agefm.isnull(), 'nevermarr'] = 1 
print (df) 
    agefm nevermarr 
0 NaN   1 
1 5.0   2 
2 6.0   3