numpy 實現(xiàn)返回指定行的指定元素的位置索引
更新時間：2021年05月10日 09:46:46 作者：徐亦快
這篇文章主要介紹了numpy 實現(xiàn)返回指定行的指定元素的位置索引操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
先上代碼，主要語句為np.where(b[c]==1)，
詳細解釋如下：
import numpy as np
b = np.array([[-2,-3,0,0,0,6,4,1],[88,1,0,0,0,6,4,2],[99,6,0,0,1,6,4,2]]) # 三行八列的數(shù)組b
print('b\n',b)
c = np.array([2,0]) # c表示指定行
print('b[c]\n',b[c]) # b[c]返回 數(shù)組b的指定行 這里依次返回了b的下標為2和0的行
print('\n')
print(np.where(b[c]==1)) # 返回指定行的指定元素的位置索引 這里返回了b[c]每行中 值為1的位置索引
觀察np.where()的返回值(array([0, 1], dtype=int64), array([4, 7], dtype=int64))可以發(fā)現(xiàn)，
對于數(shù)組b[c]來說，其中值為1的位置在[0,4]以及[1,7]上，np.where()返回了一個元組，[0]的位置表示行索引，[1]表示列索引。
綜上，np.where(b[c]==1)實現(xiàn)的功能是，從b中按c中元素的順序找到c所指定的行，然后再這些行中發(fā)現(xiàn)值為1的位置。
補充：短小精悍算例：Python如何提取數(shù)組Numpy array中指定元素的位置索引
數(shù)組a的表達式如下：
import numpy as np a = np.array([1, 2, 3, 2, 2, 3, 0, 8, 3]) print(a)
輸出結(jié)果：
[1 2 3 2 2 3 0 8 3]
現(xiàn)在要找出a中元素3的位置索引。
r1 = np.where(a==3) print(type(r1)) print(r1)
輸出結(jié)果：
<class 'tuple'>
(array([2, 5, 8], dtype=int64),)
可見得到的是一個tuple。
如果要想得到array，采用如下表示：
r1 = np.where(a==3) print(type(r1[0])) print(r1[0])
輸出結(jié)果：
<class 'numpy.ndarray'>
[2 5 8]
還可以使用argwhere()函數(shù)進行操作：
r2 = np.argwhere(a==3) print(type(r2)) print(r2) print(r2.shape)
輸出結(jié)果：
<class 'numpy.ndarray'>
[[2]
[5]
[8]]
(3, 1)
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
相關(guān)文章
在本篇文章里小編給大家整理了關(guān)于Python生成pdf目錄書簽的實例方法，有需要的朋友們可以學習下。2020-10-10
Python使用textract實現(xiàn)從各種文件中提取文本信息
textract是一個強大的Python庫,可以用于從各種文件格式中提取文本,本文將介紹textract的使用場景,以及一些常用的Python代碼案例,希望對大家有所幫助2024-01-01
這篇文章主要介紹了Django框架之django admin的命令行知識,文中通過示例代碼給大家介紹的非常詳細，需要的朋友可以參考下2021-09-09
python 的numpy庫中的mean()函數(shù)用法介紹
這篇文章主要介紹了python 的numpy庫中的mean()函數(shù)用法介紹，具有很好對參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-03-03
這篇文章主要介紹了Python日志logging模塊功能與用法,結(jié)合實例形式詳細分析了Python日志logging模塊的基本功能、原理、用法及操作注意事項,需要的朋友可以參考下2020-04-04
今天小編就為大家分享一篇Tensorflow分批量讀取數(shù)據(jù)教程，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
利用Python-iGraph如何繪制貼吧/微博的好友關(guān)系圖詳解
這篇文章主要給大家介紹了關(guān)于利用Python-iGraph如何繪制貼吧/微博好友關(guān)系圖的相關(guān)資料，文中顯示介紹了在windows系統(tǒng)下安裝python-igraph的步驟，然后通過示例代碼演示了繪制好友關(guān)系圖的方法，需要的朋友可以參考下。2017-11-11
在本篇文章里小編給大家整理了一篇關(guān)于python匿名函數(shù)的實例用法，有興趣的朋友們可以學習參考下。2021-03-03
FFmpeg是業(yè)內(nèi)有名的開源圖像視頻處理程序,在許多視頻剪輯軟件、圖像處理軟件中,都使用的FFmpeg,還有比如OBS這樣的導播軟件里面也使用了FFmpeg,FFmpeg的功能十分強大,遠不止視頻壓縮的功能,本文介紹使用FFmpeg來無損壓縮視頻文件的操作方法,感興趣的朋友一起看看吧2023-12-12
在本篇文章里小編給大家整理了關(guān)于python程序編譯相關(guān)的知識點內(nèi)容，有興趣的朋友們參考學習下。2020-06-06