在numpy数组的索引范围内置零

时间:2018-10-26 14:00:26

标签: python-3.x numpy for-loop

我有一个列表my_tuple,其中包含条件为True的所有情况的索引。我想使每个索引的所有值都为±2行,等于0。

我写了下面的代码,效果很好,但是我想用一种更加Python的方式来表达它,以避免两个for循环,运行起来更快,更易于阅读。

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

有什么想法吗?

0 个答案:

没有答案