Python - 通过交集获取两个 Pandas DataFrames 之间的列

  • Z时代
  • 2024-01-10
  • 分类：综合

要通过交集获取两个 DataFrame 之间的列，请使用该intersection()方法。让我们创建两个数据帧 -

# 创建数据框1
dataFrame1 = pd.DataFrame({"Car": ['Bentley', 'Lexus', 'Tesla', 'Mustang', 'Mercedes', 'Jaguar'],"Cubic_Capacity": [2000, 1800, 1500, 2500, 2200, 3000],"Reg_Price": [7000, 1500, 5000, 8000, 9000, 6000],
})
# 创建数据框2
dataFrame2 = pd.DataFrame({"Car": ['BMW', 'Lexus', 'Tesla', 'Mustang', 'Mercedes', 'Jaguar'],"Units_Sold": [ 100, 110, 150, 80, 200, 90]
})

获取公共列 -

dataFrame2.columns.intersection(dataFrame1.columns)

示例

以下是完整的代码 -

import pandas as pd
# 创建数据框1
dataFrame1 = pd.DataFrame({"Car": ['Bentley', 'Lexus', 'Tesla', 'Mustang', 'Mercedes', 'Jaguar'],"Cubic_Capacity": [2000, 1800, 1500, 2500, 2200, 3000],"Reg_Price": [7000, 1500, 5000, 8000, 9000, 6000],
})
print"Dataframe1...\n",dataFrame1
# 创建数据框2
dataFrame2 = pd.DataFrame({"Car": ['BMW', 'Lexus', 'Tesla', 'Mustang', 'Mercedes', 'Jaguar'],"Units_Sold": [ 100, 110, 150, 80, 200, 90]
})
print"Dataframe2...\n",dataFrame2
# 使用intersection()获取公共列
res = dataFrame2.columns.intersection(dataFrame1.columns)
print"\nCommon columns...\n",res
输出结果

这将产生以下输出 -

Dataframe1...
        Car   Cubic_Capacity   Reg_Price
0   Bentley             2000        7000
1     Lexus             1800        1500
2     Tesla             1500        5000
3   Mustang             2500        8000
4  Mercedes             2200        9000
5    Jaguar             3000        6000
Dataframe2...
        Car   Units_Sold
0       BMW          100
1     Lexus          110
2     Tesla          150
3   Mustang           80
4  Mercedes          200
5    Jaguar           90
Common columns...
Index([u'Car'], dtype='object')

以上是 Python - 通过交集获取两个 Pandas DataFrames 之间的列 的全部内容， 来源链接： utcz.com/z/350437.html

回到顶部