佳木斯湛栽影视文化发展公司

主頁 > 知識庫 > numpy 實現(xiàn)返回指定行的指定元素的位置索引

numpy 實現(xiàn)返回指定行的指定元素的位置索引

熱門標簽:企業(yè)做大做強 呼叫中心市場需求 硅谷的囚徒呼叫中心 百度AI接口 客戶服務(wù) Win7旗艦版 電話運營中心 語音系統(tǒng)

先上代碼,主要語句為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)驗,希望能給大家一個參考,也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方,望不吝賜教。

您可能感興趣的文章:
  • python如何獲得list或numpy數(shù)組中最大元素對應(yīng)的索引
  • 淺談python已知元素,獲取元素索引(numpy,pandas)
  • numpy:找到指定元素的索引示例
  • 如何獲取numpy的第一個非0元素索引

標簽:山西 喀什 安康 崇左 海南 濟南 長沙 山西

巨人網(wǎng)絡(luò)通訊聲明:本文標題《numpy 實現(xiàn)返回指定行的指定元素的位置索引》,本文關(guān)鍵詞  ;如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問題,煩請?zhí)峁┫嚓P(guān)信息告之我們,我們將及時溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò),涉及言論、版權(quán)與本站無關(guān)。
  • 相關(guān)文章
  • 收縮
    • 微信客服
    • 微信二維碼
    • 電話咨詢

    • 400-1100-266
    深水埗区| 扶绥县| 普陀区| 吉安市| 灵丘县| 武安市| 马关县| 泗水县| 平谷区| 万安县| 卫辉市| 政和县| 永新县| 湛江市| 宝丰县| 朝阳县| 鹿泉市| 青铜峡市| 翁牛特旗| 通辽市| 措勤县| 北碚区| 桐庐县| 阳泉市| 平果县| 徐汇区| 三台县| 闽侯县| 吴忠市| 浑源县| 武夷山市| 宣威市| 康保县| 巴彦县| 阿巴嘎旗| 云梦县| 英德市| 宿松县| 大宁县| 湄潭县| 九江县|