数据结构
网络
RDBMS
操作系统
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 编程
C++
C#
MongoDB
MySQL
Javascript
PHP
物理
化学
生物
数学
英语
经济
心理学
社会科学
时尚研究
法律研究Python Pandas –使用 concat() 获取两个 DataFrames 中的公共行
要获取两个 DataFrames 的公共行，请使用concat() 函数。我们用两列创建 DataFrame1 −
dataFrame1 = pd.DataFrame(
{
"Car": ['BMW', 'Lexus', 'Audi', 'Tesla', 'Bentley', 'Jaguar'],
"Reg_Price": [1000, 1500, 1100, 800, 1100, 900] }
)用两列创建 DataFrame2 −
dataFrame2 = pd.DataFrame(
{
"Car": ['BMW', 'Lexus', 'Audi', 'Tesla', 'Bentley', 'Jaguar'],
"Reg_Price": [1200, 1500, 1000, 800, 1100, 1000]
}
)使用 concat() 查找两个 DataFrames 中的公共行 −
dfRes = pd.concat([dataFrame1, dataFrame2])
重置索引 −
dfRes = dfRes.reset_index(drop=True)
分组列 −
dfGroup = dfRes.groupby(list(dfRes.columns))
获取每一行的长度来计算数量。如果数量大于 1，那么表示公共行 −
res = [k[0] for k in dfGroup.groups.values() if len(k) > 1]
示例
以下为代码 −
import pandas as pd
# Create DataFrame1
dataFrame1 = pd.DataFrame(
{
"Car": ['BMW', 'Lexus', 'Audi', 'Tesla', 'Bentley', 'Jaguar'],
"Reg_Price": [1000, 1500, 1100, 800, 1100, 900] }
)
print"DataFrame1 ...\n",dataFrame1
# Create DataFrame2
dataFrame2 = pd.DataFrame(
{
"Car": ['BMW', 'Lexus', 'Audi', 'Tesla', 'Bentley', 'Jaguar'],
"Reg_Price": [1200, 1500, 1000, 800, 1100, 1000]
}
)
print"\nDataFrame2 ...\n",dataFrame2
# finding common rows between two DataFrames
dfRes = pd.concat([dataFrame1, dataFrame2])
# reset index
dfRes = dfRes.reset_index(drop=True)
# groupby columns
dfGroup = dfRes.groupby(list(dfRes.columns))
# length of each row to calculate the count
# if count is greater than 1, that would mean common rows
res = [k[0] for k in dfGroup.groups.values() if len(k) > 1]
print"\nCommon rows...\n",dfRes.reindex(res)
输出
这样会得到以下输出 −
DataFrame1 ... Car Reg_Price 0 BMW 1000 1 Lexus 1500 2 Audi 1100 3 Tesla 800 4 Bentley 1100 5 Jaguar 900 DataFrame2 ... Car Reg_Price 0 BMW 1200 1 Lexus 1500 2 Audi 1000 3 Tesla 800 4 Bentley 1100 5 Jaguar 1000 Common rows... Car Reg_Price 3 Tesla 800 1 Lexus 1500 4 Bentley 1100
更新于: 2021-09-14
470 阅读量
广告