python - 从文本文件中读取特定行作为 numpy 数组

Question

我正在尝试读取他格式的 txt 文件:

[text]
[text]
[text]
1 0 
4 5
3 0 0
[text]
.
.
.

我需要将第 4 行到第 6 行读取为一个 numpy 数组。到目前为止，我有:

 lines=[]
 with open('filename', "r") as f:
     for i, line in enumerate(f):
         if i>=3 and i<=5:
             lines.append(line)
 lines = np.array(lines)

这会将每个必需的行作为一个元素读取，但我需要在单独的列中将数字作为单独的元素。有解决办法吗？

谢谢

Answer 1

您需要将字符串转换为整数:

lines=[]
with open('filename', "r") as f:
    for i, line in enumerate(x.split('\n')):
        if i>=3 and i<=5:
            lines.append([int(y) for y in line.split()])

lines = np.array(lines)
print type(lines)