极客教程如何在Pandas DataFrame中找到共同元素？
要在Pandas DataFrame中找到共同元素，我们可以使用带有列列表的merge（）方法。
步骤
-
创建一个二维的、大小可变的、可能是异构的表格数据, df1 。
-
打印输入的DataFrame, df1 。
-
创建另一个二维表格数据, df2 。
-
打印输入的DataFrame, df2 。
-
使用 merge() 方法查找共同元素。
-
打印共同的DataFrame。
例子
import pandas as pd
df1 = pd.DataFrame(
{
"x": [5, 2, 7, 0],
"y": [4, 7, 5, 1],
"z": [9, 3, 5, 1]
}
)
df2 = pd.DataFrame(
{
"x": [5, 2, 7, 0, 11, 12],
"y": [4, 7, 5, 1, 19, 20],
"z": [9, 3, 5, 1, 29, 30]
}
)
print("Input DataFrame 1 is:\n", df1)
print("Input DataFrame 2 is:\n", df2)
common = df1.merge(df2, on=['x', 'y', 'z'])
print("Common of DataFrame 1 and 2 is: \n", common)
输出
Input DataFrame 1 is:
x y z
0 5 4 9
1 2 7 3
2 7 5 5
3 0 1 1
Input DataFrame 2 is:
x y z
0 5 4 9
1 2 7 3
2 7 5 5
3 0 1 1
4 11 19 29
5 12 20 30
Common of DataFrame 1 and 2 is:
x y z
0 5 4 9
1 2 7 3
2 7 5 5
3 0 1 1