根据列表的条件在 DataFrame 中创建新列

Question

我有一个像这样的 DataFrame：

df = pd.DataFrame({'A' : (1,2,3), 'B': ([0,1],[3,4],[6,0])})

我想创建一个名为 test 的新列，如果 B 列的每个列表中都存在 0，则显示 1。结果可能如下所示：

df = pd.DataFrame({'A' : (1,2,3), 'B': ([0,1],[3,4],[6,0]), 'test': (1,0,1)})

对于包含字符串而不是列表的数据框，我使用以下方法从字符串值创建了额外的列

df.loc[df['B'].str.contains(0),'test']=1

当我尝试使用示例 df 时，我生成了一个

TypeError: first argument must be string or compiled pattern

我也尝试将列表转换为字符串，但只保留了列表中的第一个整数。我怀疑我需要其他东西来访问列表中的元素，但不知道该怎么做。有什么建议吗？

Answer 1

这应该为您完成：

df['test'] = pd.np.where(df['B'].apply(lambda x: 0 in x), 1, 0)

Create new column in DataFrame from a conditional of a list