要比较pandas数据框的列与集合,可以使用以下代码示例中的解决方法:
import pandas as pd
# 创建示例数据框
data = {'A': [1, 2, 3, 4, 5],
'B': [6, 7, 8, 9, 10],
'C': [11, 12, 13, 14, 15]}
df = pd.DataFrame(data)
# 创建示例集合
set1 = {2, 4, 6, 8, 10}
# 比较列与集合
# 使用isin()方法检查列中的元素是否存在于集合中,并将结果存储在新列中
df['A_in_set1'] = df['A'].isin(set1)
df['B_in_set1'] = df['B'].isin(set1)
df['C_in_set1'] = df['C'].isin(set1)
# 打印结果
print(df)
在这个例子中,我们创建了一个包含三个列(A,B和C)的数据框,并创建了一个示例集合(set1)。然后,我们使用isin()
方法检查每个列中的元素是否存在于集合中,并将结果存储在新的列(A_in_set1,B_in_set1和C_in_set1)中。最后,我们打印数据框以查看结果。
输出结果如下所示:
A B C A_in_set1 B_in_set1 C_in_set1
0 1 6 11 False True False
1 2 7 12 True False False
2 3 8 13 False True False
3 4 9 14 True False False
4 5 10 15 False True False
在输出结果中,True表示对应的元素在集合中,False表示对应的元素不在集合中。