numpy 獲取array中陣列元素的索引位置

2021-08-20 03:17:31 字數 1670 閱讀 7528

numpy - 獲取array中陣列元素的索引

2023年08月05日 10:36:59

1. 函式原型

argwhere(array):找到非空陣列array在滿足某些條件下的索引,返回索引陣列。

2. 應用

2.1 一維陣列

返回乙個一維陣列,代表當前滿足條件的元素出現的位置。

[python]view plain

copy

print?

# -*- coding: utf-8 -*-

import numpy as np  

arr = np.random.randint(0,10, (5,))  

index = np.argwhere(arr < 5)  

# -*- coding: utf-8 -*-

import numpy as np

arr = np.random.randint(0,10, (5,))

index = np.argwhere(arr < 5)

2. 2 二維陣列

返回二維陣列,代表當前滿足條件的元素出現的位置。

[python]view plain

copy

print?

# -*- coding: utf-8 -*-

import numpy as np  

」「」arr = 

9 3 7 0 

3 4 2 4 

3 6 4 4 

index = 

0   1

0   3

1   0

1   1

1   2

1   3

2   0

2   2

2   3

」「」arr = np.random.randint(0,10, (3,4))  

index = np.argwhere(arr < 5)  

# -*- coding: utf-8 -*-

import numpy as np

"""arr =

9 3 7 0

3 4 2 4

3 6 4 4

index =

0 1

0 3

1 0

1 1

1 2

1 3

2 0

2 2

2 3

"""arr = np.random.randint(0,10, (3,4))

index = np.argwhere(arr < 5)

參考文獻

python

numpy

個人分類:

python

所屬專欄:

python

numpy中array和asarray的區別

array和asarray都可以將結構資料轉化為ndarray,但是主要區別就是當資料來源是ndarray時,array仍然會copy出乙個副本,占用新的記憶體,但asarray不會。例子1 import numpy as np example 1 data1 1,1,1 1,1,1 1,1,1 a...

論numpy中matrix 和 array的區別

論numpy中matrix 和 array的區別 csdn部落格 2014年03月07日 16 26 55 numpy matrices必須是2維的,但是 numpy arrays ndarrays 可以是多維的 1d,2d,3d nd matrix是array的乙個小的分支,包含於array。所以...

numpy中matrix 和 array的區別

numpy matrices必須是2維的,但是 numpy arrays ndarrays 可以是多維的 1d,2d,3d nd matrix是array的乙個小的分支,包含於array。所以matrix 擁有array的所有特性。所以 在sklearn 中如果傳入的特徵只有乙個,要使用x.valu...