从索引和值点创建矩阵

Question

我想读取一个包含矩阵的值的文本文件。假设您有一个如下所示的.txt文件：

0 0 4.0
0 1 5.2
0 2 2.1
1 0 2.1
1 1 2.9
1 2 3.1

这里，第一列给出了x轴上矩阵的索引，第二列给出了y轴上的索引。第三列是矩阵中此位置的值。当缺少值时，该值仅为零。

我很清楚这样的事实，像.mtx格式这样的数据格式是存在的，但我想单独从这个txt文件创建一个scipy稀疏矩阵或numpy数组，而不是将它调整为.mtx文件格式。有没有一个Python函数可以帮我做这件事，而我没有呢？

Answer 1

import numpy

with open('filename.txt','r') as f:
    lines = f.readlines()
f.close()

data = [i.split(' ') for i in lines]
z = list(zip(*data))

row_indices = list(map(int,z[0]))
column_indices = list(map(int,z[1]))
values = list(map(float,z[2]))

m = max(row_indices)+1
n = max(column_indices)+1

p = max([m,n])

A = numpy.zeros((p,p))
A[row_indices,column_indices]=values

print(A)

如果您想要一个正方形矩阵，其中第一列的最大行数为行数，第二列的最大行数为大小，那么您可以删除p = max([m,n])，并用A = numpy.zeros((m,n))替换A = numpy.zeros((p,p))。

Answer 2

从按照示例中的第一列(主要列)和第二列(次要列)排序的数组(a)开始，您可以使用reshape

# a = np.loadtxt('filename')
x = len(np.unique(a[:,0]))
y = len(np.unique(a[:,1]))
a[:,2].reshape(x,y).T

输出：

array([[4. , 2.1],
       [5.2, 2.9],
       [2.1, 3.1]])