NaTを使ったPandas条件文

Question

私は多くの変数を持つデータフレームを持っています。インデックスはuidで、その他の変数はすべて日付です。

特定の値がNaTのときにフラグ変数を作成しようとしていますが、正しいステートメントが見つかりません。

私はこのような何かしたい：

auxData['flagInvited'] = np.where(auxData['invited'] == NaT, 0, 1) 

どのように私はこれを簡単に行うことができますか？

Answer 1

私はあなたがboolean maskに変換するためにnotnullが必要だと思うし、その後intにキャスト - Trueは1あるとFalseは0です：

auxData['flagInvited'] = auxData['invited'].notnull().astype(int) 

サンプル：

auxData = pd.DataFrame({'invited':[np.nan, '2017-01-01','2017-03-03']}) 
auxData.invited = pd.to_datetime(auxData.invited, dayfirst=True) 
print (auxData) 
    invited 
0  NaT 
1 2017-01-01 
2 2017-03-03 

print (auxData['invited'].notnull()) 
0 False 
1  True 
2  True 
Name: invited, dtype: bool 

auxData['flagInvited'] = auxData['invited'].notnull().astype(int) 
print (auxData) 
    invited flagInvited 
0  NaT   0 
1 2017-01-01   1 
2 2017-03-03   1