将 csv 文件读入字典

Question

我的代码如下所示：

from sys import argv
import csv

person = {}

file = argv[1]
sequence = (open(argv[2], 'r').read()).strip()

with open(file, 'r', newline = '') as csvfile:
    reader = csv.reader(csvfile)
    for k, v0, v1, v2, v3, v4, v5, v6, v7 in reader:
        person[k] = [v0, v1, v2, v3, v4, v5, v6, v7]

有没有其他方法可以遍历列以更简洁或更有效地添加到字典中？ 例如，如果您不知道列数？

Answer 1

您可以使用列表切片来实现这一点：使用行中的第一个元素作为键，其余元素作为值。

with open(file, 'r', newline = '') as csvfile:
    for row in csv.reader(csvfile):
        person[row[0]] = row[1:]

Answer 2

您可以使用unpacking ：

with open(file, 'r', newline = '') as csvfile:
    reader = csv.reader(csvfile)
    for key, *values in reader:
        person[key] = values

Answer 3

您还可以将csv文件直接读入 Pandas 数据帧：

import pandas as pd

data = pd.read_csv("filename.csv")

然后放入字典

data.to_dict()