用特定值填充numpy数组

Question

我有一个5x5的零数组。 将numpy导入为np

x = np.zeros((5,5)):

[[0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0.]]

我想在某些位置填充1：

input=[(2,2),(1,1),(3,4)] :
     [[0. 0. 0. 1. 0.]
     [0. 0. 0. 0. 0.]
     [0. 0. 1. 0. 0.]
     [0. 1. 0. 0. 0.]
     [0. 0. 0. 0. 0.]] 

在给定输入的情况下我该怎么做？

Answer 1

x = np.zeros((5,5))
for pair in input:
    x[pair[0]][pair[1]]=1

代码不言自明。

Answer 2

您可以使用use项目集

[x.itemset(t, 1) for t in input]

在元组中设置位置，最后一个参数是要设置的值