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()- 但不太灵活，这对您的情况无关紧要：

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

旁注：最好不要称您为 variable array，因为这也是 NumPy 使用的名称：这使得在执行from numpy import *or之后粘贴代码不太方便from pylab import *（在 Python shell 中），因为人们可以期望array表示numpy.array()，并且因为变量名会影响 NumPy 的array().