python - 读取csv文件并在python中转换为大写

Question

我正在尝试使用 csv 库读取 csv 文件。我试图遍历每一行，但在尝试对每一行执行一些有用的函数之前将其转换为大写。

import csv
reader = csv.DictReader(open('sample.csv', 'rb'))
for line in reader:
    line = line.upper()
    name = line['Name']

上面的代码似乎不起作用。当我尝试将读入的行转换为上部时，它失败了。我可以将读取的 csv 文件的每一列单独更改为上层(使用字典键)，但我想避免这种情况，因为字典中有很多键。

Answer 1

您可以使用生成器表达式预处理文件对象中的每一行，然后将它们传递给 csv.DictReader ，它返回一个可迭代的字典:

with open('sample.csv', 'rb') as f:
    header = next(f).strip().split(',')
    reader = csv.DictReader((l.upper() for l in f), fieldnames=header)
    for line in reader:
        name = line['Name']

<小时/>

演示:

>>> f = ['a,b', 'c,d']
>>> c = csv.DictReader((l.upper() for l in f), fieldnames=('h1', 'h2'))
>>> list(c)
[{'h2': 'B', 'h1': 'A'}, {'h2': 'D', 'h1': 'C'}]