将 csv.DictReader 与切片一起使用

Question

我想知道是否有一种方法可以使用 csv.DictReader 将除第一个列之外的所有列作为整数读取，有点像这样：

filename = sys.argv[1]
database = []
with open(filename) as file:
    reader = csv.DictReader(file)
    for row in reader:
        row[1:] = int(row[1:])
        database.append(row)

我知道这不是正确的方法，因为它给出了无法散列切片的错误。 我有办法完全避免这样做，但为了将来参考，我很好奇，是否使用切片，我可以有选择地与一行中的列进行交互，而无需对每一列进行硬编码？

Answer 1

用这个：

import csv

filename = 'test.csv'
database = []
with open(filename) as file:
    reader = csv.DictReader(file)
    for row in reader:
        new_d = {}                                     # Create new dictionary to be appended into database
        for i, (k, v) in enumerate(row.items()):       # Loop through items of the row (i = index, k = key, v = value)
                new_d[k] = int(v) if i > 0 else v      # If i > 0, add int(v) to the dictionary, else add v
        database.append(new_d)                         # Append to database
                
print(database)

测试.csv：

Letter,Num1,Num2
A,123,456
B,789,012
C,345,678
D,901,234
E,567,890

输出：

[{'Letter': 'A', 'Num1': 123, 'Num2': 456},
 {'Letter': 'B', 'Num1': 789, 'Num2': 12},
 {'Letter': 'C', 'Num1': 345, 'Num2': 678},
 {'Letter': 'D', 'Num1': 901, 'Num2': 234},
 {'Letter': 'E', 'Num1': 567, 'Num2': 890}]

Answer 2

您可以通过使用key()字典方法来获取每个字典中的键列表以及用于进行转换的切片：

import csv
from pprint import pprint
import sys


filename = sys.argv[1]
database = []
with open(filename) as file:
    reader = csv.DictReader(file)
    for row in reader:
        for key in list(row.keys())[1:]:
            row[key] = int(row[key])
        database.append(row)

pprint(database)

输出：

[{'name': 'John', 'number1': 1, 'number2': 2, 'number3': 3, 'number4': 4},
 {'name': 'Alex', 'number1': 4, 'number2': 3, 'number3': 2, 'number4': 1},
 {'name': 'James', 'number1': 1, 'number2': 3, 'number3': 2, 'number4': 4}]