【发布时间】:2018-09-20 16:09:04
【问题描述】:
我有一个存储阈值的 numpy 数组和另一个存储一些值的数组。我希望这些最后的值小于或等于相应的阈值。特别是,如果一个值大于其相应的阈值,我应该用阈值更改它。
以下示例正是我想要的,但我想知道是否存在更好的方法来实现它,或者是否已经存在一个 numpy 方法(我搜索过但我没有找到)来做到这一点。
In [1]: import numpy as np
In [2]: a = np.random.rand(10)
In [3]: a
Out[3]:
array([0.38331068, 0.32042463, 0.89980916, 0.86472908, 0.10812789,
0.35855107, 0.09916983, 0.55710449, 0.38591185, 0.70798023])
In [4]: t = np.array([0.95, 0.9, 0.8, 0.75, 0.7, 0.65, 0.6, 0.55, 0.5, 0.45])
In [5]: mask = a > t
In [6]: a[mask] = t[mask]
In [7]: a
Out[7]:
array([0.38331068, 0.32042463, 0.8 , 0.75 , 0.10812789,
0.35855107, 0.09916983, 0.55 , 0.38591185, 0.45 ])
【问题讨论】:
标签: arrays python-3.x numpy