如何将文件逐行存储到 numpy.ndarray (Python)

Question

我必须导入一个文件，例如包含 3 行数字：

1 2 3 4
1 2 3
1 2 3 4 5 6

如何将它们存储到numpy.ndarray M中，例如M[0]给出一个包含第一行的np.array ，即[1,2,3,4] ？

提前致谢。

Answer 1

这个怎么样：

import numpy as np


with open('test_file.txt') as file:
    arr = np.array(
        [
            np.array([float(num) for num in line.strip().split("\t")])
            for line in file
        ]
    )
print(arr)

并且该数组应如下所示：

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

Answer 2

与 Giorgios 相同，但使用 np.array 而不是行列表：

import numpy as np


with open('test_file.txt') as file:
    arr = np.array(
        [
            np.array([float(num) for num in line.split(" ")])
            for line in file
        ]
    )
print(arr)