时间戳

首页 > 解决方案 > 从文本文件中读取矩阵并将其存储在数组中

python - 从文本文件中读取矩阵并将其存储在数组中

问题描述

定义

我将以下矩阵存储在文本文件中：

 1 0 0 1 0 1 1
 0 1 0 1 1 1 0
 0 0 1 0 1 1 1

我想从文本图块中读取这个矩阵，并使用 python 2.7 将它存储在一个二维数组中。

我尝试的代码

我尝试的代码如下：

f = open('Matrix.txt')
triplets=f.read().split()
for i in range(0,len(triplets)): triplets[i]=triplets[i].split(',')
A = np.array(triplets, dtype=np.uint8)

print(A)

问题

就目前而言，上面的代码是以一维方式打印矩阵。是否可以按照上面矩阵中的定义以二维方式保存矩阵？

标签： pythonnumpymatrixtext-filesbinary-data

解决方案

使用np.loadtxt：

A = np.loadtxt('filename.txt')

>>> A
array([[ 1.,  0.,  0.,  1.,  0.,  1.,  1.],
       [ 0.,  1.,  0.,  1.,  1.,  1.,  0.],
       [ 0.,  0.,  1.,  0.,  1.,  1.,  1.]])

或者，您可以像您正在做的那样逐行阅读它（但这不是有效的）：

A = []
with open('filename.txt', 'r') as f:
    for line in f:
        A.append(list(map(int,line.split())))

>>> np.array(A)
array([[1, 0, 0, 1, 0, 1, 1],
       [0, 1, 0, 1, 1, 1, 0],
       [0, 0, 1, 0, 1, 1, 1]])

推荐阅读