在python中查找索引和更改值的最佳方法

Question

我有一个numpy.ndarray X.我想找到X中X＆gt;阈值的所有位置，然后X =阈值。

这样做的最便宜的方式（就时间复杂性而言）是什么？我需要运行这个程序数百万次。谢谢！

Answer 1

据我所知，您可以使用numpy索引来替换大于某个阈值的所有元素。

虽然，我不确定这是最快的方式。

threshold = 10 # for example
some_array[some_array > threshold] = threshold

Answer 2

尝试numpy.where：

    from numpy import where

    Y  = where( X> treshold, threshold,X)

where以及ufunc-way中的if语句 where（condition，if True，else）