numpy array找出符合條件的數(shù)并賦值的示例代碼
更新時間：2022年05月31日 14:57:47 作者：hitrjj
本文主要介紹了numpy array找出符合條件的數(shù)并賦值的示例代碼，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
在python中利用numpy array進(jìn)行數(shù)據(jù)處理，經(jīng)常需要找出符合某些要求的數(shù)據(jù)位置，有時候還需要對這些位置重新賦值。這里總結(jié)了幾種找出符合條件數(shù)據(jù)位置的方法。
這里以一個8*8的隨機(jī)數(shù)組舉例,來找出大于零的數(shù)。
import numpy as np
a = random.randint(-10,10,size=(8,8))
>>>
array([[ 5, 5, -7, 7, -8, -7, 0, -8],
[ -4, 9, 8, -3, 6, -4, -7, -5],
[ 7, 0, 6, 6, -4, -2, -8, 2],
[ 6, -5, 8, 4, 7, -8, -4, -4],
[ 0, 1, -1, -8, -1, 9, 4, 1],
[ 4, -8, -1, -8, -2, -6, -1, 9],
[ 7, 7, 9, -9, 4, 8, 3, 1],
[ -8, 4, -2, 4, -1, -4, -10, 0]])
1.直接利用條件索引
location= a[a>0] print(location) >>> array([5, 5, 7, 9, 8, 6, 7, 6, 6, 2, 6, 8, 4, 7, 1, 9, 4, 1, 4, 9, 7, 7, 9, ? ? ? ?4, 8, 3, 1, 4, 4]) # 直接輸出了大于0的數(shù)字 #--------------------------------------------------------------# # 我們可以用下面的方法將小于0的數(shù)字都設(shè)置為零，留下大于零的數(shù)字 b = a.copy() b[b<=0]=0 print(b) >>> [[5 5 0 7 0 0 0 0] ?[0 9 8 0 6 0 0 0] ?[7 0 6 6 0 0 0 2] ?[6 0 8 4 7 0 0 0] ?[0 1 0 0 0 9 4 1] ?[4 0 0 0 0 0 0 9] ?[7 7 9 0 4 8 3 1] ?[0 4 0 4 0 0 0 0]] ?# 這就將所有大于零的保留了下來 #--------------------------------------------------------------# #還可以此類推，將大于零的位置都設(shè)置成1,可得到大于一的位置 b = a.copy() b[b>0] = 1 b[b<=0] = 0 print(b) >>> [[1 1 0 1 0 0 0 0] ?[0 1 1 0 1 0 0 0] ?[1 0 1 1 0 0 0 1] ?[1 0 1 1 1 0 0 0] ?[0 1 0 0 0 1 1 1] ?[1 0 0 0 0 0 0 1] ?[1 1 1 0 1 1 1 1] ?[0 1 0 1 0 0 0 0]]
2.利用numpy.where()
# results = np.where(condition, [x, y]) # 當(dāng)條件為真時，對應(yīng)位置返回x中的值，條件不成立則返回y中的值 c = np.where(a>0,a,0) ?#滿足大于0的值保留，不滿足的設(shè)為0 print(c) >>> [[5 5 0 7 0 0 0 0] ?[0 9 8 0 6 0 0 0] ?[7 0 6 6 0 0 0 2] ?[6 0 8 4 7 0 0 0] ?[0 1 0 0 0 9 4 1] ?[4 0 0 0 0 0 0 9] ?[7 7 9 0 4 8 3 1] ?[0 4 0 4 0 0 0 0]] # 大于零為1小于零為0 c = np.where(a>0,1,0) ?#滿足大于0的值保留，不滿足的設(shè)為0 print(c) [[1 1 0 1 0 0 0 0] ?[0 1 1 0 1 0 0 0] ?[1 0 1 1 0 0 0 1] ?[1 0 1 1 1 0 0 0] ?[0 1 0 0 0 1 1 1] ?[1 0 0 0 0 0 0 1] ?[1 1 1 0 1 1 1 1] ?[0 1 0 1 0 0 0 0]]
3.直接邏輯運(yùn)算
a > 0 ? # 得到判斷矩陣 array([[ True, ?True, False, ?True, False, False, False, False], ? ? ? ?[False, ?True, ?True, False, ?True, False, False, False], ? ? ? ?[ True, False, ?True, ?True, False, False, False, ?True], ? ? ? ?[ True, False, ?True, ?True, ?True, False, False, False], ? ? ? ?[False, ?True, False, False, False, ?True, ?True, ?True], ? ? ? ?[ True, False, False, False, False, False, False, ?True], ? ? ? ?[ True, ?True, ?True, False, ?True, ?True, ?True, ?True], ? ? ? ?[False, ?True, False, ?True, False, False, False, False]], dtype=bool)
到此這篇關(guān)于numpy array找出符合條件的數(shù)并賦值的示例代碼的文章就介紹到這了,更多相關(guān)numpy array賦值內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了Python決策樹之基于信息增益的特征選擇,結(jié)合實(shí)例形式分析了決策樹中基于信息增益的特征選擇原理、計(jì)算公式、操作流程以及具體實(shí)現(xiàn)技巧,需要的朋友可以參考下2018-06-06
這篇文章主要為大家詳細(xì)介紹了pyQt4實(shí)現(xiàn)俄羅斯方塊游戲，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-06-06
Python使用騰訊云API實(shí)現(xiàn)短信驗(yàn)證碼功能
使用Python與騰訊云接口對接,實(shí)現(xiàn)短信驗(yàn)證碼功能變得非常簡單,只需要幾行代碼就能夠輕松實(shí)現(xiàn)短信的發(fā)送,無須關(guān)心復(fù)雜的短信協(xié)議和底層實(shí)現(xiàn),讀者可以根據(jù)自己的實(shí)際需求,靈活使用騰訊云短信SDK提供的API來實(shí)現(xiàn)更豐富的短信功能2024-01-01
python實(shí)現(xiàn)系統(tǒng)狀態(tài)監(jiān)測和故障轉(zhuǎn)移實(shí)例方法
這篇文章主要介紹了用python實(shí)現(xiàn)系統(tǒng)狀態(tài)監(jiān)測和故障轉(zhuǎn)移的代碼方法2013-11-11
這篇文章主要介紹了python第三方庫pygame的使用,pygame一般用來做游戲，在這需要注意在使用pygame提供的功能之前，需要調(diào)用init方法，本文給大家介紹的非常詳細(xì)，需要的朋友可以參考下2022-08-08
Python基于火山引擎豆包大模型搭建QQ機(jī)器人詳細(xì)教程(2024年最新)
這篇文章主要介紹了Python基于火山引擎豆包大模型搭建QQ機(jī)器人詳細(xì)的相關(guān)資料,包括開通模型、配置APIKEY鑒權(quán)和SDK安裝等步驟,文中通過圖文介紹的非常詳細(xì),需要的朋友可以參考下2025-01-01
PDF批量加密?是一種保護(hù)PDF文件安全性的方法,通過為多個PDF文件設(shè)置相同的密碼,防止未經(jīng)授權(quán)的用戶訪問這些文件,下面我們來看看如何使用Python制作一個PDF批量加密工具吧2025-01-01
這篇文章主要介紹了python之?dāng)M合的實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-07-07
這篇文章主要介紹了安裝PyInstaller失敗問題解決,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-12-12
python實(shí)現(xiàn)redis三種cas事務(wù)操作
本篇文章主要介紹了python實(shí)現(xiàn)redis三種cas事務(wù)操作，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-12-12