python 判断csv某一列文本数据是否含有特定词,若含有,则根据不同特定词取不同值,如,2,3,
时间: 2023-10-09 19:07:09 浏览: 106
假设我们有一个名为`data.csv`的文件,其中包含以下数据:
```
name,age,gender
Tom,25,male
Jane,30,female
Bob,28,male
```
我们想要判断`gender`列是否含有特定词`male`或`female`,如果含有,则根据不同特定词取不同值。
下面是一个实现的例子:
```python
import csv
# 定义一个字典,用于存储不同特定词对应的值
values = {'male': 2, 'female': 3}
# 打开 csv 文件
with open('data.csv') as f:
# 读取 csv 文件
reader = csv.DictReader(f)
# 遍历每一行数据
for row in reader:
# 判断 gender 列是否含有特定词
if 'male' in row['gender']:
print(row['name'], values['male'])
elif 'female' in row['gender']:
print(row['name'], values['female'])
```
输出结果为:
```
Tom 2
Jane 3
Bob 2
```
在上面的例子中,我们首先定义了一个字典`values`,用于存储不同特定词对应的值。然后打开`data.csv`文件,使用`csv.DictReader`读取文件,遍历每一行数据。对于每一行数据,我们判断`gender`列是否含有特定词`male`或`female`,如果含有,则根据不同特定词取不同值,并输出结果。
阅读全文