python – 将0和1从文本文件转换为2-D数组

我得到一个只有0和1值的文本文件.一个例子是:

010
111
010

我需要取这些值并将它们变成二维数组.所以对于这个例子,我应该得到一个这样的数组:

[[0,1,0] , [1,1,1], [0,1,0]]

每个文本文件都有不同的列数和宽度,因此我的程序应该能够处理不同的输入矩阵大小.我尝试过以下方法:

numpy.loadtxt("file1.txt")

但是,这给了我以下输出数组:

[  10.  111.   10.]

任何指导都会很棒!

最佳答案
你不需要numpy.readtext.只需使用内置文件处理程序:

import numpy as np
with open('file1.txt') as f:
    arr = np.array([list(map(int, line.strip())) for line in f])
print(arr)

转载注明原文:python – 将0和1从文本文件转换为2-D数组 - 代码日志