带有NaN的Pandas布尔索引

Question

我有这个玩具示例来捕捉我的真实问题：

import pandas as pd
import numpy as np
df = pd.DataFrame({'A': ['car there is','car not working', 'bus there is']})
df.iloc[1] = np.nan
idx = df['A'].str.contains('car')
df['IsCar'] = 0
df.loc[idx,'IsCar'] = 1

当我尝试运行此代码时，收到以下错误消息：

ValueError: cannot index with vector containing NA / NaN values

为什么我不能这样做。是否有解决方法我不需要用其他东西替换NaN？

Answer 1

na（see docs）有一个标记str.contains，您可以将其设置为False，这将为缺失值提供填充值。只需使用

idx = df['A'].str.contains('car', na=False)