如何将csv文件加载到矩阵python

Question

我有一个名为 TEMP2.csv 的 .CSV 文件，其中包含以下内容：

1376460059,4,33.29,33.23,33.23,33.29,33.23,33.29,33.29,33.29,33.33,33.29,33.33,33.29,33.33,33.33,33.37,33.33,33.33,33.33,33.33,33.37,33.37,33.37,33.37

这就是我阅读文件的方式：

import csv
import numpy as np

data = np.genfromtxt('TEMP2.csv',delimiter=',', skip_header=2)

跳过第一个两个数字后，我需要将接下来的 4 个数字添加到矩阵的第一行。下一个 4 到同一矩阵中的第二行，依此类推。

但是这不起作用，我对python真的很陌生，已经尝试了几天，但我仍然没有得到结果。我会非常感谢你的帮助。谢谢。

Answer 1

也许你需要reshape。如果它无法重塑，则在数组末尾添加零。

COLUMN_NUM = 4
data = np.genfromtxt('a.csv', delimiter=',')[2:]

if data.shape[0] % 4 == 0:
    print data.reshape((-1, 4))
else:
    data = np.pad(data, (0, COLUMN_NUM - len(data) % COLUMN_NUM), 'constant')
    print data.reshape((-1, COLUMN_NUM))

新的形状应该和原来的形状兼容。所以它reshape成功，你会得到这样的结果：

[[ 33.29  33.23  33.23  33.29]
 [ 33.23  33.29  33.29  33.29]
 [ 33.33  33.29  33.33  33.29]
 [ 33.33  33.33  33.37  33.33]
 [ 33.33  33.33  33.33  33.37]
 [ 33.37  33.37  33.37  33.11]]

希望这会有所帮助。