python 判断csv某一列数据是否含有特定词,若含有,则根据不同特定词取不同值,如,2,3,
时间: 2023-10-08 19:04:42 浏览: 46
可以使用Python的csv模块来读取csv文件,然后遍历每一行数据,判断特定列是否包含特定词,根据不同特定词取不同值。
以下是一个示例代码:
```python
import csv
filename = 'data.csv' # csv文件名
target_col = 2 # 目标列
special_words = ['apple', 'banana', 'orange'] # 特定词列表
with open(filename, newline='') as csvfile:
reader = csv.reader(csvfile)
for row in reader:
if row[target_col] in special_words:
# 根据不同特定词取不同值
if row[target_col] == 'apple':
value = 2
elif row[target_col] == 'banana':
value = 3
elif row[target_col] == 'orange':
value = 4
# 处理取到的值
print(value)
```
在以上示例代码中,我们使用了csv模块中的reader函数来读取csv文件,并使用一个特定列的索引来获取该列的数据。然后,我们遍历每一行数据,判断目标列的数据是否在特定词列表中,如果是,则根据不同特定词取不同值,并进行处理。
阅读全文