Numpy从位置矩阵和值矩阵构造矩阵

Question

我有一个Nx3矩阵

i   j  val
.   .   .
.   .   .
.   .   .

我想从中构造一个矩阵，其中元素x_ij=val，例如位置值矩阵

[[ 0 0 3 ] 
 [ 1 1 2 ] 
 [ 0 1 4 ]
 [ 1 0 1 ]
 [ 2 0 5 ]
 [ 2 1 2 ]]

会产生矩阵

[[3 4]
 [1 2]
 [5 2]]   

在numpy中，最简单/简洁的方法是什么？你知道吗

Answer 1

创建一个空数组并用索引值填充它

In [541]: x = np.array([[0, 0, 3],
     ...:                      [1, 1, 2],
     ...:                      [0, 1, 4],
     ...:                      [1, 0, 1],
     ...:                      [2, 0, 5],
     ...:                      [2, 1, 2]])
     ...:                      
In [543]: arr = np.zeros((3,2),int)
In [544]: arr[x[:,0], x[:,1]] = x[:,2]
In [545]: arr
Out[545]: 
array([[3, 4],
       [1, 2],
       [5, 2]])

在本例中，定义了arr的所有元素，但即使有些元素保留为0，也可以这样做。如果存在重复项，并且希望对它们求和，那么coo方法很方便。如果x不完整，reshape建议将出现问题。你知道吗