【发布时间】:2018-07-18 12:51:17
【问题描述】:
我有一个包含多列的数据框,
data={'NAME':['A','B','C'],
'A':[[1,2,3],[np.nan],[3,4,5]],
'B':[[2,3,4],[3,4,5],[np.nan]],
'C':[[2,4],[3],[6,7]] }
df=pd.DataFrame(data)
df=df[['A','B','C']]
我可以使用列的交集,
df.assign(D=df.apply(
lambda x: list(set(x.A).intersection(set(x.B)).intersection(set(x.C))),
axis=1))
这将创建一个 D 列,
D
[2]
[]
[]
但我想忽略具有 na 值的单元格并获得交集, 预期输出是,
D
[2]
[3]
[]
如何做到这一点? 感谢您的帮助。
【问题讨论】: