以下是一个示例代码,用于比较数据框中的两列是否匹配,并创建一个包含匹配项的新数据框:
import pandas as pd
# 创建示例数据框
data = {'Column1': [1, 2, 3, 4, 5],
'Column2': [1, 2, 6, 4, 9]}
df = pd.DataFrame(data)
# 比较两列是否匹配
matches = df['Column1'] == df['Column2']
# 创建新数据框,只包含匹配项
new_df = df[matches]
# 打印新数据框
print(new_df)
输出结果为:
Column1 Column2
0 1 1
1 2 2
3 4 4
在这个示例中,我们首先创建了一个包含两列数据的数据框。然后,我们使用比较运算符(==
)比较两列是否匹配,并将结果存储在一个布尔序列中。最后,我们使用布尔序列作为索引来选择匹配项,并创建一个新的数据框。