判断csv的第三列数据是否含有某些关键词,第三列是文本句子,如果含有则取值1,否则取值0,并将结果保存
时间: 2024-05-02 15:19:24 浏览: 222
为新的一列。
可以使用Python中的csv模块和字符串的in关键字来实现。
示例代码:
```python
import csv
# 定义关键词列表
keywords = ['apple', 'banana', 'orange']
# 打开csv文件
with open('data.csv', newline='') as csvfile:
# 读取csv文件内容
reader = csv.reader(csvfile)
# 获取表头
header = next(reader)
# 添加新的一列
header.append('含有关键词')
# 打开新的csv文件
with open('result.csv', 'w', newline='') as resultfile:
# 写入表头
writer = csv.writer(resultfile)
writer.writerow(header)
# 遍历每一行数据
for row in reader:
# 判断第三列数据是否含有关键词
if any(keyword in row[2] for keyword in keywords):
row.append(1)
else:
row.append(0)
# 写入新的一行数据
writer.writerow(row)
```
以上代码会读取名为data.csv的csv文件,并判断第三列数据是否含有关键词列表中的任一关键词,如果含有则在该行数据后添加1,否则添加0,并将结果保存为名为result.csv的新文件。
阅读全文