python - 将列表添加到 numpy 数组

Question

现在我正在编写一个从文件中读取数据的函数，目标是将这些数据添加到 numpy 数组并返回所述数组。

我想将数组作为二维数组返回，但是我不确定数组的完整形状是什么(我知道列数，但不知道行数)。我现在拥有的是:

columns = _____
for line in currentFile:
     currentLine = line.split()
     data = np.zeros(shape=(columns),dtype=float)
     tempData = []
     for i in range(columns):
          tempData.append(currentLine[i])
      data = np.concatenate((data,tempdata),axis=0)

但是，这会生成一个一维数组。

本质上我要问的是:

有没有办法将 python 列表作为行添加到行数可变的 numpy 数组中？

Answer 1

如果您的文件data.txt是

1 2 3 4
1 2 3 4
1 2 3 4
1 2 3 4
1 2 3 4
1 2 3 4

您需要做的就是

>>> import numpy as n
>>> data_array = n.loadtxt("data.txt")
>>> data_array

array([[1., 2., 3., 4.],
       [1., 2., 3., 4.],
       [1., 2., 3., 4.],
       [1., 2., 3., 4.],
       [1., 2., 3., 4.],
       [1., 2., 3., 4.]])