1. 程式人生 > >python np.where()

python np.where()

numpy.where()函式是三元表示式x if condition else y的向量化版本。假設我們有一個布林陣列和兩個值陣列:

x = np.array([1.1, 1.2, 1.3, 1.4, 1.5])
y = np.array([2.1, 2.2, 2.3, 2.4, 2.5])
condition = np.array([True, False, True, True, False])
假設我們想要根據condition中的值選取x和y的值:當condition中的值為True時,選取x的值,否則從y中選取。
result = np.where(condition, x, y)
列印的結果為:【1.1, 2.2, 1.3, 1.4, 2.5】