在numpy ndarray中查找并替换特定值

Question

我想通过numpy ndarray迭代，如果任何值小于x，请用x。

替换其中一个

我尝试过array_name[ array_name < X] = X，但这取代了小于x的所有值。

我可以使用循环，但是我觉得可能已经与numpy捆绑在一起的更简洁的方法。

        for i in array_name:
            if i < X:
                i = X
                break

有没有办法更优雅？

Answer 1

array_name < X

返回相同的数组，但以真或错误返回。然后，您可以选择单元格为true

的索引

idx = np.argwhere(array_name < X)[i]
array_name[idx] = value

在这里，您可以选择i任意选择