對numpy中的where方法嵌套使用詳解

今天小編就爲大家分享一篇對numpy中的where方法嵌套使用詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

如同for循環一樣，numpy中的where方法可以實現嵌套功能。這是簡化嵌套式矩陣邏輯的一個很好的方法。

假設有一個矩陣，需要把小於0的元素改成-1，大於0的元素改成1，而等於0的時候不做修改。

那麼，對應的代碼示範如下：

#!/usr/bin/python
 
import numpy as np
 
data = np.random.randn(4,5)
data1 =np.where(data > 0,1,
np.where(data <0,-1,0))
print("datavalue:")
print(data)
print("data1value:")
print(data1)

程序的執行結果如下：

In [3]: %runpython_exp04.py

data value:

[[-2.06262429 0.94548656 -0.29458562 0.82657 -1.08587439]
 [-0.67416161 0.77247191 0.60330603 0.73694198 -0.63761278]
 [ 0.24887356 -0.27086027 0.34312363 0.727303 0.72741593]
 [-0.48973095 -0.33185631 -1.23341695 0.13569267 2.06881178]]

data1 value:

[[-1 1 -1 1-1]
 [-1 1 1 1 -1]
 [ 1 -1 1 1 1]
 [-1 -1 -1 1 1]]

這種用法，不僅比單純通過複合的循環遍歷實現的代碼更加簡潔，而且在執行效率上有着很大的優勢。很多時候，計算的向量化就是爲了能夠加速整個程序的執行速度。

以上這篇對numpy中的where方法嵌套使用詳解就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持神馬文庫。