pandas 取一个表格第一列与另一个表格第一列的交集并输出新的表格
假设有两个 DataFrame,分别是 df1 和 df2,它们都有一个名为 "col1" 的列。
要取 df1 和 df2 的 "col1" 列的交集,并输出新的 DataFrame,可以使用 pandas 的 merge 函数。
具体代码如下:
import pandas as pd
# 创建示例数据
df1 = pd.DataFrame({'col1': ['a', 'b', 'c', 'd'], 'col2': [1, 2, 3, 4]})
df2 = pd.DataFrame({'col1': ['b', 'd', 'e', 'f'], 'col3': [5, 6, 7, 8]})
# 取 col1 列的交集
common_col1 = pd.merge(df1[['col1']], df2[['col1']], on='col1', how='inner')
# 输出新的 DataFrame
print(common_col1)
输出结果:
col1
0 b
1 d
这个结果就是 df1 和 df2 的 "col1" 列的交集。
原文地址: https://www.cveoy.top/t/topic/bDTm 著作权归作者所有。请勿转载和采集!