读取csv文件时如何创建字典？

Question

我有以下蛮力代码：

import csv
d = {}
with open(csv_filename, 'r') as csv_file:
    reader = csv.reader(csv_file)
    for row in reader:
        key = row[0]
        d[key] = np.array(
            [[float(x) for x in row[1:4]], [float(x) for x in row[4:7]],
             [float(x) for x in row[7:]]], dtype=float)

有没有一种方法可以重写它以避免[1:4], [4:7]等？

Answer 1

如果我理解正确，则只需要重塑数据即可。

key = row[0]
d[key] = np.reshape(np.array(row).astype(float), (3, 3))