Быстрый способ создания массива numpy 1d bool с известными ненулевыми индексами записи - PullRequest
1 голос
/ 12 июля 2020

Учитывая 1d np.ndarray, содержащий список индексов, который составляет True: [1, 2, 4], и длину цели np.ndarray: 6

Как мы можем быстро построить фактическое np.ndarray, которое должно быть [False, True, True, False, True, False]

1 Ответ

2 голосов
/ 12 июля 2020
idx = [1,2,3]
s = 6
a = np.zeros(s,dtype=bool)
a[idx] = True

вывод:

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