如何在 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

更新于：07-Sep-2021

5K+ 浏览次数

开启你的 职业道路

完成课程以获得认证

开始学习
广告