Python vlookup
时间: 2023-08-14 18:34:17 浏览: 156
Python Excel vlookup函数实现过程解析
5星 · 资源好评率100%
在Python中实现类似于Excel中的vlookup功能,可以使用pandas库中的merge函数。
首先需要导入pandas库:
```python
import pandas as pd
```
假设有两个数据表,一个是左表"table1",一个是右表"table2",它们的共同列是"key"。现在需要在"table1"中添加一列"new_column",该列的值来自"table2"中的"lookup_column"列,根据"key"列进行匹配。
代码如下:
```python
# 创建左表
table1 = pd.DataFrame({'key': ['A', 'B', 'C', 'D'],
'value': [1, 2, 3, 4]})
# 创建右表
table2 = pd.DataFrame({'key': ['B', 'D', 'E', 'F'],
'lookup_column': ['value2', 'value4', 'value5', 'value6']})
# 使用merge函数实现vlookup
result = pd.merge(table1, table2, on='key', how='left')
result['new_column'] = result['lookup_column']
# 输出结果
print(result)
```
输出结果:
```
key value lookup_column new_column
0 A 1 NaN NaN
1 B 2 value2 value2
2 C 3 NaN NaN
3 D 4 value4 value4
```
可以看到,"table1"中的"A"和"C"没有匹配到"table2"中的对应值,因此"new_column"列的值为NaN。而"B"和"D"成功匹配到了对应的值,"new_column"列的值等于"lookup_column"列的值。
阅读全文