Сделать нули в диапазоне индексов в массиве NumPy - PullRequest
0 голосов
/ 26 октября 2018

У меня есть список my_tuple, в котором есть индексы всех случаев, когда условие истинно.Я хочу сделать равным 0, все значения ± 2 строки из каждого индекса.

Я написал следующий код, который отлично работает, но я хотел бы выразить его более питоническим способом, чтобы избежать двухдля циклов, чтобы работать быстрее и облегчить чтение.

for i in range(0, my_tuple[0].shape[0]):
    for j in range(my_tuple[0][i] - 2, my_tuple[0][i] + 3):
        if j < column.shape[0]:
            column[j] = 0

Есть идеи?

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