編程速記(17)：numpy篇-對np.array做閾值限定

一、場景描述

如果我們希望對一個np.array做閾值限制，即令大於某一個數的元素賦值爲a，否則則賦值爲b

二、demo

import numpy as np 
a = np.asarray([1,2,3])
print(a>1)
a[a>1] = 0
print(a)

運行結果

[False  True  True]
[1 0 0]
[Finished in 0.5s]