判断csv某一列是否含有关键词,直接改当前csv数据,有则取值1,没有则取值0
时间: 2024-05-09 13:21:30 浏览: 13
以下是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=''参数,以避免出现额外的空行。
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![](https://img-home.csdnimg.cn/images/20210720083646.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)