先上代碼,主要語句為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]]) # 三行八列的數組b
print('b\n',b)
c = np.array([2,0]) # c表示指定行
print('b[c]\n',b[c]) # b[c]返回 數組b的指定行 這里依次返回了b的下標為2和0的行
print('\n')
print(np.where(b[c]==1)) # 返回指定行的指定元素的位置索引 這里返回了b[c]每行中 值為1的位置索引
![](/d/20211017/725bb71276aa1d686158be0cc4c75862.gif)
觀察np.where()的返回值(array([0, 1], dtype=int64), array([4, 7], dtype=int64))可以發(fā)現,
對于數組b[c]來說,其中值為1的位置在[0,4]以及[1,7]上,np.where()返回了一個元組,[0]的位置表示行索引,[1]表示列索引。
綜上,np.where(b[c]==1)實現的功能是,從b中按c中元素的順序找到c所指定的行,然后再這些行中發(fā)現值為1的位置。
補充:短小精悍算例:Python如何提取數組Numpy array中指定元素的位置索引
數組a的表達式如下:
import numpy as np
a = np.array([1, 2, 3, 2, 2, 3, 0, 8, 3])
print(a)
輸出結果:
[1 2 3 2 2 3 0 8 3]
現在要找出a中元素3的位置索引。
r1 = np.where(a==3)
print(type(r1))
print(r1)
輸出結果:
class 'tuple'>
(array([2, 5, 8], dtype=int64),)
可見得到的是一個tuple。
如果要想得到array,采用如下表示:
r1 = np.where(a==3)
print(type(r1[0]))
print(r1[0])
輸出結果:
class 'numpy.ndarray'>
[2 5 8]
還可以使用argwhere()函數進行操作:
r2 = np.argwhere(a==3)
print(type(r2))
print(r2)
print(r2.shape)
輸出結果:
class 'numpy.ndarray'>
[[2]
[5]
[8]]
(3, 1)
以上為個人經驗,希望能給大家一個參考,也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方,望不吝賜教。
您可能感興趣的文章:- python如何獲得list或numpy數組中最大元素對應的索引
- 淺談python已知元素,獲取元素索引(numpy,pandas)
- numpy:找到指定元素的索引示例
- 如何獲取numpy的第一個非0元素索引