以下是一个示例代码,演示如何比较DataFrame的列与另一个列表,并进行修改:
import pandas as pd
# 创建示例DataFrame
df = pd.DataFrame({'A': [1, 2, 3, 4, 5],
'B': [6, 7, 8, 9, 10],
'C': [11, 12, 13, 14, 15]})
# 创建示例列表
lst = [1, 3, 5]
# 比较DataFrame的列与列表,并进行修改
df['A'] = df['A'].isin(lst)
# 打印修改后的DataFrame
print(df)
输出结果为:
A B C
0 True 6 11
1 False 7 12
2 True 8 13
3 False 9 14
4 True 10 15
在上述示例中,我们首先创建了一个DataFrame对象df和一个列表lst。然后,我们使用isin()
方法将DataFrame列"A"与列表进行比较,并将结果保存回同一列。最后,我们打印修改后的DataFrame对象。