numpy 陣列索引陣列

2022-08-20 15:06:09 字數 1081 閱讀 7847

在numpy中,陣列除了可以被整數索引,還可以被陣列索引.

a[b]就是已陣列b的元素為索引,讀取陣列a的值.

當被索引陣列a是一維陣列,b是一維或則多維陣列時,結果維度維度與索引陣列b相同。

a = np.array([7,8,9,10])

b=np.array([[3,1],[1,2]])

print('a:'

,a)print('b:'

,b)print('

result:

',a[b])

print(a[b].shape)

a: [ 7  8  9 10]

b: [[3 1]

[1 2]]

result: [[10 8]

[ 8 9]]

(2, 2)

當被索引陣列a是多維陣列,b是一維或則多維陣列時,每乙個唯一的索引數列指向a的第一維。

a =np.array([

[0, 0, 0], #黑色

[255, 0, 0], #

紅色 [0, 255, 0], #

綠色 [0, 0, 255], #

藍色 [255, 255, 255] #

白色])

b=np.array([

[0, 1, 2, 0],

[0, 3, 4, 0]

])print

(a.shape)

print

(b.shape)

print('

result:

',a[b])

print(a[b].shape)

(5, 3)

(2, 4)

result: [[[ 0 0 0]

[2550 0]

[ 0 2550]

[ 0 0 0]]

[[ 0 0 0]

[ 0 0 255]

[255 255 255]

[ 0 0 0]]]

(2, 4, 3)

numpy陣列的索引

numpy陣列的索引 個人學習筆記,分享給大家!一 索引的作用 獲取 修改陣列單個元素的值 二 獲取單個元素的值 1,在一維陣列中 import numpy as np x np.array 5,3,6,9,8 x 0 索引為非負值,獲取第i個值,從0開始計數 out 8 5 x 1 索引為負值,從...

Numpy布林陣列索引

布林陣列索引 布林陣列索引允許選擇陣列的任意元素。通常,這種型別的索引用於選擇滿足某些條件的陣列元素。下面是乙個例子 import numpy as np a np.array 1,2 3,4 5,6 找出a中大於2的元素 這將返回乙個與a相同形狀的布林值的numpy陣列,其中的bool idx的每...

Numpy陣列索引與切片 高階索引

有一些numpy中索引結果時候不太理解的地方,以後可以多多熟悉,熟悉後加以運用。import numpy as np a np.array 1,2,3 3,4,5 4,5,6 print a 1 第2列元素 print a 1,第2行元素 print a 1 第2列及剩下的所有元素整數陣列的索引 以...