python-3.x - Pandas:检查一列是否存在于另一列中

Question

我是 Python 和 Pandas 的新手。我有一个具有以下结构的数据集。这是一只 Pandas DF

city time1              time2
a    [1991, 1992, 1993] [1993,1994,1995]

time1 和 time2 代表两个来源的数据覆盖率。我想创建一个新列，指示 time1 和 time2 是否有任何交集，如果有则返回 True，否则返回 False。这项任务听起来非常简单。我正在考虑在两列上使用集合操作，但它没有按预期工作。谁能帮我解决这个问题？

谢谢!

感谢您的帮助。

Answer 1

您可以遍历所有列并将列表更改为集合，并查看交集中是否有任何值。

df1 = df.applymap(lambda x: set(x) if type(x) == list else set([x]))
df1.apply(lambda x: bool(x.time1 & x.time2), axis=1)

这是一种半向量化的方式，应该可以让它运行得更快。

df1 = df[['time1', 'time2']].applymap(lambda x: set(x) if type(x) == list else set([x]))
(df1.time1.values & df1.time2.values).astype(bool)

甚至更快

change_to_set = lambda x: set(x) if type(x) == list else set([x])
time1_set = df.time1.map(change_to_set).values
time2_set = df.time2.map(change_to_set).values
(time1_set & time2_set).astype(bool)