利用python处理csv文件根据字典进行替换
时间: 2023-03-06 14:04:11 浏览: 332
python处理csv文件
可以使用Python中的csv模块和字典对象来处理CSV文件并根据字典进行替换。具体步骤如下:
1. 使用csv模块读取CSV文件,可以使用csv.reader函数来读取每行数据并返回一个列表对象。
2. 创建一个字典对象,将需要替换的内容作为字典的key,替换后的内容作为value。
3. 遍历CSV文件中的每行数据,将需要替换的内容通过字典的get()方法获取替换后的值,替换掉原始的内容。
4. 将替换后的数据写回到CSV文件中,可以使用csv.writer函数将数据写入文件。
以下是一个简单的示例代码:
```python
import csv
# 读取CSV文件
with open('example.csv', 'r') as f:
reader = csv.reader(f)
rows = list(reader)
# 创建字典对象
replace_dict = {
'apple': 'orange',
'cat': 'dog',
'red': 'blue'
}
# 遍历CSV文件中的每行数据,进行替换
for i, row in enumerate(rows):
for j, cell in enumerate(row):
if cell in replace_dict:
rows[i][j] = replace_dict[cell]
# 将替换后的数据写回到CSV文件中
with open('example_new.csv', 'w', newline='') as f:
writer = csv.writer(f)
writer.writerows(rows)
```
在上面的代码中,我们首先读取了名为`example.csv`的CSV文件,然后创建了一个字典对象`replace_dict`来存储需要替换的内容。接着,我们遍历了CSV文件中的每行数据,如果某个单元格中的内容在字典对象中存在,则将其替换成字典中对应的值。最后,我们将替换后的数据写回到了名为`example_new.csv`的文件中。
阅读全文