從numpy數(shù)組中取出滿足條件的元素示例
例如問題：從 arr 數(shù)組中提取所有奇數(shù)元素。
input：arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) output: #> array([1, 3, 5, 7, 9]) Solution: #Input >>> arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) #Solution arr = arr[arr % 2 == 1] >>> array([1, 3, 5, 7, 9])
這是對奇數(shù)元素的提取，其他條件同理，在這個問題上我們深入再學習一下，首先 numpy 中的 array 數(shù)組是可以直接對數(shù)組中的每個元素進行處理的。例如我們想對每個元素減一：
#input >>> arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) #Solution >>> arr array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) >>> arr - 1 array([-1, 0, 1, 2, 3, 4, 5, 6, 7, 8])
從上面的例子我們可以看出來，對 arr 直接進行減一操作，結(jié)果是對每一個元素都進行了減一的操作，數(shù)組的維度不變，為了滿足取出特定條件的元素這個操作，我們可以進行條件判斷，例如判斷每個元素是否為奇數(shù)：
#input >>> arr = np.arange(10) >>> arr array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) #solution >>> arr % 2 == 1 array([False, True, False, True, False, True, False, True, False, True])
由此可見，條件判斷后輸出為和 arr 相同維度的數(shù)組，但是其中每個元素為布爾類型的值，取值為數(shù)組 arr 對每個元素分別進行條件判斷的結(jié)果。
下面我們分析對 arr 數(shù)組進行特定取值，給其一個布爾類型的 arr 數(shù)組，并且維度與 arr 相同，以此得到取值的目的：
#input >>> arr_bool = np.array([1, 1, 0, 0, 1, 1, 0, 0, 1, 1], dtype=bool) >>> arr_bool array([ True, True, False, False, True, True, False, False, True, True]) #solution >>> arr[arr_bool] array([0, 1, 4, 5, 8, 9])
從上述例子，我們構(gòu)建了一個布爾類型的數(shù)組為 arr_bool ，并且維度與 arr 相同，進行直接嵌入實現(xiàn)了取值，以此完成了我們最終的目的，可以對任意滿足條件的元素進行提取。并且理解了其中的原理。
PS：番外篇，如果維度不同可以嗎？
#input >>> arr_bool = np.ones(11, dtype=bool) >>> arr_bool array([ True, True, True, True, True, True, True, True, True, True, True]) #注意??！這里有11個 True 哦。 #solution >>> arr[arr_bool] Traceback (most recent call last): File "<stdin>", line 1, in <module> IndexError: boolean index did not match indexed array along dimension 0; dimension is 10 but corresponding boolean dimension is 11
由上可見，維度必須保持一致，從 arr 數(shù)組里才可以按照 bool 類型取出對應位置為 True 的元素。這個性質(zhì)幫助我們可以更靈活的處理 numpy 數(shù)組。
PS2: 如何替換滿足條件的元素為其它值呢？
>>> arr = np.arange(10) >>> arr array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) >>> arr[arr%2 == 1] = -1 >>> arr array([ 0, -1, 2, -1, 4, -1, 6, -1, 8, -1])
以上這篇從numpy數(shù)組中取出滿足條件的元素示例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
100行Python代碼實現(xiàn)自動搶火車票（附源碼）
又到年底了，相信對于在外地的朋友們來說，火車票是到年底最頭痛的一件事了，但作為程序員的你怎么能一樣呢？快發(fā)揮你的特長，下面這篇文章主要給大家介紹了如果通過100行Python代碼實現(xiàn)自動搶火車票的相關資料，需要的朋友可以參考下。2018-01-01
這篇文章主要介紹了Python 代碼實現(xiàn)各種酷炫功能，生成二維碼、生成詞云、批量摳圖、文字情緒識別等功能分享，需要的小伙伴可以參考一下2022-03-03
這篇文章主要介紹了Python繪制交通流折線圖詳情，文章基于python的相關資料展開折線圖繪制的實現(xiàn)流程，感興趣的小伙伴可以參考一下2022-06-06
使用 Python 進行 PDF 文檔處理可以通過多種庫來實現(xiàn),包括 PyPDF2、pdfplumber、reportlab、pdfminer 等,這些庫可以處理不同的 PDF 任務,以下是幾種常見操作及對應的庫和代碼示例,感興趣的小伙伴跟著小編一起來看看吧2024-09-09
利用插件和python實現(xiàn)Excel轉(zhuǎn)json的兩種辦法
轉(zhuǎn)換Excel表格到JSON格式有很多方法,下面這篇文章主要給大家介紹了關于利用插件和python實現(xiàn)Excel轉(zhuǎn)json的兩種辦法,文中通過代碼介紹的非常詳細,需要的朋友可以參考下2023-11-11
這篇文章主要介紹了Python中的文件定位方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08
pandas如何將dataframe中的NaN替換成None
這篇文章主要介紹了pandas如何將dataframe中的NaN替換成None問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08
這篇文章主要為大家介紹了Python編寫漏洞驗證腳本來批量測試繁瑣的漏洞實現(xiàn)，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步2022-02-02
Python3.4實現(xiàn)從HTTP代理網(wǎng)站批量獲取代理并篩選的方法示例
這篇文章主要介紹了Python3.4實現(xiàn)從HTTP代理網(wǎng)站批量獲取代理并篩選的方法,涉及Python網(wǎng)絡連接、讀取、判斷等相關操作技巧,需要的朋友可以參考下2017-09-09
結(jié)合Python的SimpleHTTPServer源碼來解析socket通信
SimpleHTTPServer是Python中一個現(xiàn)成的HTTP服務器例子,本文我們將結(jié)合Python的SimpleHTTPServer源碼來解析socket通信,我們先來看一下socket的基本概念:2016-06-06