pandas 模仿vlookup查找关联的两列数据
时间: 2023-08-23 13:20:52 浏览: 106
你可以使用 pandas 中的 merge() 方法来实现类似于 vlookup 的操作。假设你有两个数据框 df1 和 df2,它们都有一个共同的列名为 key,你想要将它们关联起来并查找相关的数据。
代码如下:
```
import pandas as pd
# 创建两个数据框
df1 = pd.DataFrame({'key': ['A', 'B', 'C', 'D'],
'value': [1, 2, 3, 4]})
df2 = pd.DataFrame({'key': ['B', 'D', 'E', 'F'],
'value': [5, 6, 7, 8]})
# 使用 merge() 方法将两个数据框关联起来
result = pd.merge(df1, df2, on='key')
print(result)
```
输出:
```
key value_x value_y
0 B 2 5
1 D 4 6
```
在这个例子中,我们将 df1 和 df2 关联起来,并查找它们之间 key 列相同的数据。merge() 方法将两个数据框按照 key 列进行匹配,并返回一个新的数据框 result,其中包含了两个数据框中 key 列相同的数据以及它们的其他列。在这个例子中,我们得到了一个包含了 key、value_x 和 value_y 列的新数据框,其中 value_x 列来自 df1,value_y 列来自 df2。
阅读全文