NumPy 切片和索引

Posted 2021-03-20 daicw

import numpy as np
 
a = np.arange(10,20,1)
b = slice(1,5,2)
d = a[1:5:2]
e = a[2:]
print(a)
print(a[b])
print(d)
print(e)
#结果
[10 11 12 13 14 15 16 17 18 19]
[11 13]
[11 13]
[12 13 14 15 16 17 18 19]

import numpy as np
 
a = np.array([[1,2,3],[3,4,5],[4,5,6]])
print(a)
# 从某个索引处开始切割
print(‘从数组索引 a[1:] 处开始切割‘)
print(a[1:])
print(‘省略号 …‘)
print (a[...,1])   # 第2列元素
print (a[1,...])   # 第2行元素
print (a[...,1:])  # 第2列及剩下的所有元素
print (a[...,0:2])  # 第1,2列所有元素
#结果

[[1 2 3]
 [3 4 5]
 [4 5 6]]
从数组索引 a[1:] 处开始切割
[[3 4 5]
 [4 5 6]]
省略号 …
[2 4 5]
[3 4 5]
[[2 3]
 [4 5]
 [5 6]]
[[1 2]
 [3 4]
 [4 5]]