python - 从 .csv 文件创建一个 numpy 数组，但我只能得到数组中的第一行——我错过了什么？

Question

我的文件中有这样的数据

0.5,0,-21
0.5,0,-21
0.5,0,-19
0.5,0,-20
0.5,0,-20
1,0,-31
1,0,-28
1,0,-31
1,0,-28
1,0,-30

我想创建一个 numpy 数组。我正在执行以下操作以尝试将其复制到数组中:

with open ("bot1.csv") as fd:
    array = numpy.fromfile(fd, count=-1, dtype=float, sep=",")

但结果数组只是:

array([  0.5,   0. , -21. ])

知道我做错了什么吗？

Answer 1

我建议您使用 numpy.loadtxt() .它比 genfromtxt() 更快——但灵 active 较低，这对您的情况应该无关紧要:

table = numpy.loadtxt('bot1.csv', delimiter=',')

旁注:最好不要称你为变量 array，因为这也是 NumPy 使用的名称:这使得在执行 from numpy import * 后粘贴代码不方便 或 from pylab import *(在 Python shell 中)，因为人们可以期望 array 表示 numpy.array()，并且因为变量名称隐藏了 NumPy 的 array()。