python - 将 CSV 转换为 numpy

Question

我的 CSV 中有以下数据:

column 1    column 2
    5         17
    9         16
    4         13
   10         15
    2         10
    1         18
    2         14
    1         19

我正在尝试从 CSV 到 dict(这部分已经正确完成)，然后从 dict 到 numpy 以获得中间价格。但是，我收到以下错误:

Traceback (most recent call last):File "untitled.py", line 10, in <module>
avg = np.median(row['column 1']). 

TypeError: cannot perform reduce with flexible type

如何编写此代码才能正常工作？

import csv
import numpy as np

storedata = []

f = open('untitled.csv', 'r')
reader = csv.DictReader(f)
for row in reader:
    col_1 = int(row['column 1'])
    avg = np.median(row['column 1'])

print(avg)

Answer 1

您的 header 有点难以解析，因为 header 名称中有空格作为分隔符和空格。最简单的方法是跳过标题:

>>> data = np.genfromtxt('mydata.csv', skip_header=True)
>>> median_col1 = np.median(data[:,0])
>>> median_col1
3.0