（Python）如何使用[：]语法对数组的每个元素使用条件语句？

Question

如果我需要对整数的numpy.ndarray中的每个元素都要求一个条件，那么我必须使用for循环吗

for i in range(n):
    if a[i] == 0:
        a[i] = 1   

或者我可以用[：]语法问这个问题吗

if a[:] == 0:
    #...

我知道前一个是错误的，但有没有办法做类似的事情？

Answer 1

假设a是数组，并且希望将大于1的a值更改为等于1：

a[a > 1] = 1

这是因为表达式a > 1创建了mask array，并且当掩码数组用作索引（它在这里）时，该操作仅适用于True索引。