如何使用标头完整的python导入csv文件，其中第一列为非数字

Question

这是上一个问题的详细说明，但是当我深入研究python时，我对python如何处理csv文件感到更加困惑。

我有一个csv文件，并且必须保持这种状态（例如，无法将其转换为文本文件）。它相当于5行乘11列的数组或矩阵或向量。

我一直在尝试使用在这里和其他地方（例如python.org）找到的各种方法来读取csv，以便它保留列和行之间的关系，其中第一行和第一列=非数值。其余为浮点值，并包含正浮点数和负浮点数。

我想做的是导入csv并在python中编译它，这样，如果我要引用列标题，它将返回存储在行中的关联值。例如：

>>> workers, constant, age
>>> workers
    w0
    w1
    w2
    w3
    constant
    7.334
    5.235
    3.225
    0
    age
    -1.406
    -4.936
    -1.478
    0

依此类推…

我正在寻找用于处理这种数据结构的技术。我是python的新手。

Answer 1

Python的csv模块按行处理数据，这是查看此类数据的常用方法。您似乎想要按列的方法。这是一种方法。

假设您的文件已命名myclone.csv并包含

workers,constant,age
w0,7.334,-1.406
w1,5.235,-4.936
w2,3.2225,-1.478
w3,0,0

此代码应该给您一个或两个想法：

>>> import csv
>>> f = open('myclone.csv', 'rb')
>>> reader = csv.reader(f)
>>> headers = next(reader, None)
>>> headers
['workers', 'constant', 'age']
>>> column = {}
>>> for h in headers:
...    column[h] = []
...
>>> column
{'workers': [], 'constant': [], 'age': []}
>>> for row in reader:
...   for h, v in zip(headers, row):
...     column[h].append(v)
...
>>> column
{'workers': ['w0', 'w1', 'w2', 'w3'], 'constant': ['7.334', '5.235', '3.2225', '0'], 'age': ['-1.406', '-4.936', '-1.478', '0']}
>>> column['workers']
['w0', 'w1', 'w2', 'w3']
>>> column['constant']
['7.334', '5.235', '3.2225', '0']
>>> column['age']
['-1.406', '-4.936', '-1.478', '0']
>>>

要将数值放入浮点数，请添加此内容

converters = [str.strip] + [float] * (len(headers) - 1)

提前做

for h, v, conv in zip(headers, row, converters):
  column[h].append(conv(v))

而不是上面类似的两行。