numpy中關(guān)于where函數(shù)的用法
關(guān)于where函數(shù)用法
在我的理解中np.where函數(shù)有三個(gè)用法
1. np.where()[0] 和 np.where()[1]
where在我的理解中是一個(gè)尋找數(shù)組中某個(gè)元素的函數(shù)
在此用法中np.where()[0] 表示行索引，np.where()[1]表示列索引
具體如下：
import numpy as np
array = np.arange(12).reshape(3,4)
print('array:', array)
print('np.where(array > 5):', np.where(array > 5))
print('array[np.where(a > 5)]:', array[np.where(array > 5)])
print('np.where(array > 5)[0]:', np.where(array > 5)[0])
print('np.where(array > 5)[1]:', np.where(array > 5)[1])
print(array[np.where(array > 5)[0], np.where(array > 5)[1]])
array: [[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
np.where(array > 5): (array([1, 1, 2, 2, 2, 2]), array([2, 3, 0, 1, 2, 3]))
array[np.where(array > 5)]: [ 6 7 8 9 10 11]
np.where(array > 5)[0]: [1 1 2 2 2 2]
np.where(array > 5)[1]: [2 3 0 1 2 3]
[ 6 7 8 9 10 11]
2. np.where(condition, x, y)
滿足condition 這個(gè)條件的輸出x，不滿足輸出y。
如下所示：
小于2的數(shù)改為0，否則改為10
import numpy as np t2=np.array([[1,2,3],[1,2,3]]) print(np.where(t2<2,0,10)) [[ 0 10 10] [ 0 10 10]]#輸出
3. np.where(condition)
只有條件 (condition)，沒(méi)有x和y，則輸出滿足條件 (即非0) 元素的坐標(biāo) (類(lèi)似于numpy.nonzero)。
舉個(gè)簡(jiǎn)單的例子：
import numpy as np t2=np.array([[1,2,3],[1,2,3]]) print(np.where(t2<2)) (array([0, 1], dtype=int64), array([0, 0], dtype=int64)) #輸出
所以 np.where會(huì)輸出每個(gè)符合要求的元素的對(duì)應(yīng)的坐標(biāo)
需要注意的一點(diǎn)是：
輸入的不能直接是list，需要轉(zhuǎn)為array或者為array才行。
比如range(10)和np.arange(10)后者返回的是數(shù)組，使用np.where才能達(dá)到效果。
總結(jié)
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
python中return的返回和執(zhí)行實(shí)例
今天小編就為大家分享一篇python中return的返回和執(zhí)行實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-12-12
Python3.6安裝卸載、執(zhí)行命令、執(zhí)行py文件的方法詳解
這篇文章主要介紹了Python3.6之安裝卸載、執(zhí)行命令、執(zhí)行py文件的方法，本文給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-02-02
python使用requests設(shè)置讀取超時(shí)時(shí)間
在Python中,使用requests庫(kù)進(jìn)行網(wǎng)絡(luò)請(qǐng)求時(shí),可以通過(guò)設(shè)置?timeout參數(shù)來(lái)指定讀取超時(shí)時(shí)間,本文就來(lái)介紹一下,具有一定的參考價(jià)值,感興趣的可以了解一下2023-11-11
這篇文章主要給大家介紹了Python中三個(gè)不可思議的返回功能,文中通過(guò)實(shí)例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2022-01-01
python實(shí)現(xiàn)自動(dòng)打卡小程序
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)自動(dòng)打卡小程序，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2021-03-03
實(shí)現(xiàn)字符倒轉(zhuǎn)的python函數(shù)2008-09-09
Python 帶你快速上手 Apache APISIX 插件開(kāi)發(fā)
Apache APISIX Python Runner 來(lái)了，社區(qū)中的小伙伴們?cè)陂_(kāi)發(fā) Apache APISIX 插件時(shí)又多了一種新選擇，本文將用實(shí)列向大家介紹,需要的朋友可以參考下面文章內(nèi)容2021-09-09
matplotlib調(diào)整子圖間距,調(diào)整整體空白的方法
今天小編就為大家分享一篇matplotlib調(diào)整子圖間距,調(diào)整整體空白的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-08-08
Python基于pandas實(shí)現(xiàn)json格式轉(zhuǎn)換成dataframe的方法
這篇文章主要介紹了Python基于pandas實(shí)現(xiàn)json格式轉(zhuǎn)換成dataframe的方法,結(jié)合實(shí)例形式分析了Python使用pandas模塊操作json數(shù)據(jù)轉(zhuǎn)換成dataframe的相關(guān)操作技巧與注意事項(xiàng),需要的朋友可以參考下2018-06-06
python實(shí)現(xiàn)循環(huán)語(yǔ)句1到100累和
這篇文章主要介紹了python循環(huán)語(yǔ)句1到100累和方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-05-05