在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]
(a.shape)
(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列及剩下的所有元素整數陣列的索引 以...