NumPy -使用genfromtxt从文件中解压不同数据类型的数据

Question

我正在尝试读入一个文件，其中第一列是日期字符串，第2-4列是常规浮点数。如果我这样做了

data = np.genfromtxt("infile.csv", delimiter=','),

我在file中得到了一个包含4x行的数组，其中列1中的所有值都是(非常正确的) NaNs。所以为了得到我的日期字符串，我试着

data = np.genfromtxt("infile.csv", delimiter=',', dtype=("|S20",float,float,float))

结果是一个一维数组，现在每行的所有四列都是该数组的一个元素。

有人能解释一下我做错了什么吗？

Answer 1

有更好的方法可以做到这一点，但由于我们不知道如何快速处理日期字符串(根据数据量快速写入，而不是运行)，因此可以使用一些循环：

data_tup = tuple(np.empty(len(data), dtype=data.dtype[i]) for i in range(len(data[0])))
for i, line in enumerate(data):
    for j, item in enumerate(line):
        data_tup[j][i] = item

这将为您提供一个ndarray元组，其中包含来自每列的数据。