заполнить массив numpy указанными c значениями - PullRequest
0 голосов
/ 12 марта 2020

У меня есть массив 5x5 с нулями. import 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.]] 

как мне сделать этот ввод?

Ответы [ 2 ]

0 голосов
/ 12 марта 2020

Вы можете использовать itemset

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

в кортеже, где вы устанавливаете местоположение, а последним аргументом является значение для установки

0 голосов
/ 12 марта 2020
x = np.zeros((5,5))
for pair in input:
    x[pair[0]][pair[1]]=1

код не требует пояснений.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...