將 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}]