python:把csv中按照skc添加一列url字段,从另一张csv表中取skc相同时添加,把url字段添加到另一张表
时间: 2024-05-01 16:18:14 浏览: 87
中的相应行,该如何实现?
这个问题可以分为两个部分来实现:
1. 从另一张csv表中取skc相同的url字段。
读取这张表,将skc和url字段存储为字典类型,key为skc,value为url。遍历第一张表中的每一行,如果该行的skc在字典中存在,则添加一个url字段,并将对应的url值赋给该字段。
2. 添加url字段到另一张表中的相应行。
在遍历第一张表时,如果该行的skc在字典中不存在,则可以不做任何操作。如果存在,则可以使用pandas库的merge方法将第二张表与第一张表进行合并,之后就可以使用第一部分的方法将url字段添加到第一张表中了。
代码示例:
```python
import pandas as pd
# 读取第一张表
df1 = pd.read_csv('table1.csv')
# 读取第二张表
df2 = pd.read_csv('table2.csv')
# 将df2转换为字典类型,key为skc,value为url
url_dict = dict(zip(df2['skc'], df2['url']))
# 遍历df1,添加url字段
for i, row in df1.iterrows():
if row['skc'] in url_dict:
df1.at[i, 'url'] = url_dict[row['skc']]
# 将url字段添加到df2中,并保存结果
result = pd.merge(df1, df2, on='skc', how='left')
result.to_csv('result.csv', index=False)
```
阅读全文