1. 程式人生 > >numpy where()函式

numpy where()函式

原文連結:https://blog.csdn.net/qq_32005671/article/details/74011993

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】