如何將文件逐行存儲到 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)