判断csv某一列是否含有关键词,直接改当前csv数据,有则取值1,没有则取值0
时间: 2024-05-09 15:21:30 浏览: 65
对于csv文件的操作
以下是Python代码实现:
```python
import csv
# 定义要检查的关键词
keywords = ['关键词1', '关键词2', '关键词3']
# 打开CSV文件并读取数据
with open('data.csv', 'r', encoding='utf-8') as csvfile:
reader = csv.reader(csvfile)
rows = list(reader)
# 判断每行数据的第二列是否含有关键词
for row in rows:
if any(keyword in row[1] for keyword in keywords):
row.append('1')
else:
row.append('0')
# 将修改后的数据写回CSV文件
with open('data.csv', 'w', encoding='utf-8', newline='') as csvfile:
writer = csv.writer(csvfile)
writer.writerows(rows)
```
以上代码假设CSV文件名为data.csv,要检查的关键词存储在列表keywords中,要检查的列为第二列。代码会打开CSV文件并读取所有数据,然后遍历每行数据,使用any()函数判断该行的第二列是否含有任意一个关键词,如果有,则在该行末尾添加'1',否则添加'0'。最后将修改后的数据写回CSV文件。注意,写入CSV文件时需要指定newline=''参数,以避免出现额外的空行。
阅读全文