python匹配两列数据加标签_基于标签在同一数据帧中查找值并添加到新列(Vlookup)...
时间: 2024-02-17 22:01:01 浏览: 19
你可以使用 Pandas 包中的 merge() 函数来实现类似于 Excel 中的 Vlookup 功能。具体步骤如下:
1. 加载数据:首先,你需要将两个数据集加载到 Pandas 数据帧中,例如:
```
import pandas as pd
df1 = pd.read_csv('data1.csv')
df2 = pd.read_csv('data2.csv')
```
其中,data1.csv 和 data2.csv 是两个包含需要匹配的数据的 CSV 文件。
2. 合并数据:接下来,你可以使用 merge() 函数将两个数据帧根据共同的列进行合并,例如:
```
merged_df = pd.merge(df1, df2, on='column_name')
```
其中,column_name 是需要匹配的共同列的名称。
3. 添加标签:现在,你可以在合并后的数据帧中添加一个新列,其中包含基于标签查找到的值,例如:
```
merged_df['new_column_name'] = merged_df.apply(lambda row: 'label1' if row['column_name2'] == 'value1' else 'label2', axis=1)
```
其中,column_name2 是需要在 data2.csv 中查找值的列的名称,value1 是需要匹配的值,label1 和 label2 是你希望添加到新列中的标签。
4. 保存数据:最后,你可以将合并后的数据帧保存到 CSV 文件中,例如:
```
merged_df.to_csv('merged_data.csv', index=False)
```
其中,merged_data.csv 是保存合并后的数据的文件名。