您好,登錄后才能下訂單哦!
如同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方法嵌套使用詳解就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持億速云。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。