我有一个列表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
有什么想法吗?