pandas匹配两组数据
时间: 2023-10-12 16:16:23 浏览: 245
Pandas 测试用数据集
可以使用pandas的merge函数来匹配两组数据。
假设有两个DataFrame,分别为df1和df2,它们的列名为A和B,需要按照A列进行匹配,得到一个新的DataFrame。
代码如下:
```python
import pandas as pd
# 构造df1和df2
df1 = pd.DataFrame({'A': [1, 2, 3, 4], 'B': ['a', 'b', 'c', 'd']})
df2 = pd.DataFrame({'A': [1, 2, 5], 'B': ['x', 'y', 'z']})
# 使用merge函数进行匹配,根据A列进行合并
df = pd.merge(df1, df2, on='A')
# 输出结果
print(df)
```
输出结果为:
```
A B_x B_y
0 1 a x
1 2 b y
```
其中,B_x和B_y分别表示合并前的df1和df2中的B列。可以使用rename函数将它们改为统一的列名。
阅读全文