使用NaT的熊猫条件声明

Question

所以，我有一个包含许多变量的数据框。索引是uid，其他变量都是日期。

我试图在某个值为NaT时创建标志变量，但我找不到正确的声明。

我想要这样的事情：

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

我怎样才能轻松完成这项工作？

Answer 1

我认为您需要notnull才能转换为boolean mask，然后转为int - True为1且False为{{ 1}}：

0

样品：

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