Python NumPy實(shí)現(xiàn)數(shù)組搜索示例詳解
搜索數(shù)組
可以在數(shù)組中搜索(檢索)某個(gè)值,然后返回獲得匹配的索引。
要搜索數(shù)組,請(qǐng)使用 where() 方法。
實(shí)例
查找值為 4 的索引:
import numpy as np arr = np.array([1, 2, 3, 4, 5, 4, 4]) x = np.where(arr == 4) print(x)
運(yùn)行實(shí)例

上例會(huì)返回一個(gè)元組:(array([3, 5, 6],)
意思就是值 4 出現(xiàn)在索引 3、5 和 6。
實(shí)例
查找值為偶數(shù)的索引:
import numpy as np arr = np.array([1, 2, 3, 4, 5, 6, 7, 8]) x = np.where(arr%2 == 0) print(x)
運(yùn)行實(shí)例

實(shí)例
查找值為奇數(shù)的索引:
import numpy as np arr = np.array([1, 2, 3, 4, 5, 6, 7, 8]) x = np.where(arr%2 == 1) print(x)
運(yùn)行實(shí)例

搜索排序
有一個(gè)名為 searchsorted() 的方法,
該方法在數(shù)組中執(zhí)行二進(jìn)制搜索,并返回將在其中插入指定值以維持搜索順序的索引。
假定 searchsorted() 方法用于排序數(shù)組。
實(shí)例
查找應(yīng)在其中插入值 7 的索引:
import numpy as np arr = np.array([6, 7, 8, 9]) x = np.searchsorted(arr, 7) print(x)
運(yùn)行實(shí)例

例子解釋: 應(yīng)該在索引 1 上插入數(shù)字 7,以保持排序順序。
該方法從左側(cè)開(kāi)始搜索,并返回第一個(gè)索引,其中數(shù)字 7 不再大于下一個(gè)值。
從右側(cè)搜索
默認(rèn)情況下,返回最左邊的索引,但是我們可以給定 side=‘right’,以返回最右邊的索引。
實(shí)例
從右邊開(kāi)始查找應(yīng)該插入值 7 的索引:
import numpy as np arr = np.array([6, 7, 8, 9]) x = np.searchsorted(arr, 7, side='right') print(x)
運(yùn)行實(shí)例

例子解釋:應(yīng)該在索引 2 上插入數(shù)字 7,以保持排序順序。
該方法從右邊開(kāi)始搜索,并返回第一個(gè)索引,其中數(shù)字 7 不再小于下一個(gè)值。
多個(gè)值
要搜索多個(gè)值,請(qǐng)使用擁有指定值的數(shù)組。
實(shí)例
查找應(yīng)在其中插入值 2、4 和 6 的索引:
import numpy as np arr = np.array([1, 3, 5, 7]) x = np.searchsorted(arr, [2, 4, 6]) print(x)
運(yùn)行實(shí)例

返回值是一個(gè)數(shù)組:[1 2 3] 包含三個(gè)索引,其中將在原始數(shù)組中插入 2、4、6 以維持順序。
到此這篇關(guān)于Python NumPy實(shí)現(xiàn)數(shù)組搜索示例詳解的文章就介紹到這了,更多相關(guān)Python NumPy數(shù)組搜索內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Pyside6開(kāi)發(fā)使用Qt?Designer的示例代碼
本文主要介紹了Pyside6開(kāi)發(fā)使用Qt?Designer的示例代碼,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2025-05-05
用python實(shí)現(xiàn)一個(gè)讓人戒不掉的百變款消消樂(lè)
消消樂(lè)的熱門程度幾乎趕上王者榮耀,你是否也有收到過(guò)好友邀請(qǐng)你幫解鎖關(guān)卡的時(shí)候,今天小編帶你用python編寫(xiě)一個(gè)自己的消消樂(lè)升級(jí)版,同學(xué)請(qǐng)往下看2021-09-09
淺談Pytorch中的自動(dòng)求導(dǎo)函數(shù)backward()所需參數(shù)的含義
今天小編就為大家分享一篇淺談Pytorch中的自動(dòng)求導(dǎo)函數(shù)backward()所需參數(shù)的含義,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-02-02
Python服務(wù)器創(chuàng)建虛擬環(huán)境跑代碼
本文主要介紹了Python服務(wù)器創(chuàng)建虛擬環(huán)境跑代碼,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2022-07-07
python web.py服務(wù)器與客戶端的實(shí)現(xiàn)示例
本文介紹了使用Python的web.py庫(kù)搭建一個(gè)簡(jiǎn)單的Web服務(wù)器,并演示了自定義端口、URL映射以及POST和GET請(qǐng)求的處理,具有一定的參考價(jià)值,感興趣的可以了解一下2025-10-10
Python實(shí)現(xiàn)多種場(chǎng)景下查找并高亮Word文檔中的文本
在處理長(zhǎng)篇 Word 文檔時(shí),快速定位關(guān)鍵信息并將其突出顯示是一項(xiàng)非常實(shí)用的技能,本文將深入探討如何使用 Python 實(shí)現(xiàn)多種場(chǎng)景下的 Word 文檔文本查找與高亮功能,有需要的小伙伴可以了解下2026-03-03
python基于itchat模塊實(shí)現(xiàn)微信防撤回
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)微信防撤回,基于itchat模塊,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2019-04-04

