將文本文件中的0和1轉換爲二維數組

Question

我給出了一個僅包含0和1值的文本文件。一個例子是：

010 
111 
010 

我需要把這些值，把它們變成一個二維數組。因此，對於本例中，我應該得到的數組是這樣的：

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

每個文本文件都有不同數量的列和寬度的，所以我的計劃應該能夠採取不同的輸入矩陣大小的照顧。我曾嘗試以下：

numpy.loadtxt("file1.txt") 

然而，這給了我下面的輸出數組：

[ 10. 111. 10.] 

任何指導意見將是巨大的！

Answer 1

你不需要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) 

Answer 2

呦也可以不用map魔法。

import numpy as np 

with open('file1.txt') as f: 
    arr = np.array([[int(bin) for bin in line] for line in f]) 
print(arr) 

Answer 3

您可以使用np.genfromtxt來獲得固定寬度的列。 delimiter參數指定每列的寬度。我使用dtype=np.int8來使用最小的尺寸，因爲OP表示的值只會是零或一。

import numpy as np 

np.genfromtxt('file1.txt', delimiter=[1, 1, 1], dtype=np.int8) 

array([[0, 1, 0], 
     [1, 1, 1], 
     [0, 1, 0]])