numpy中双冒号的作用

Posted 2021-11-30 shiyublog

 1 import numpy as np
 2 a = np.array([[1, 2, 3, 1, 3, 5], [4, 5, 6, 2, 4, 3]])
 3 print(‘a‘)
 4 print(a)
 5 print(a[:, 0::2])
 6 
 7 # a
 8 # [[1 2 3 1 3 5]
 9 #  [4 5 6 2 4 3]]
10 # [[1 3 3]
11 #  [4 6 4]]
12 
13 b = np.array([[1, 2, 3, 1, 3, 5], [4, 5, 6, 2, 4, 3]])
14 print(‘b‘)
15 print(b)
16 print(b[:, 1::2])
17 
18 # b
19 # [[1 2 3 1 3 5]
20 #  [4 5 6 2 4 3]]
21 # [[2 1 5]
22 #  [5 2 3]]

a[:, 0::2]表示取a的第0纬度的所有值，第一纬度从第0列开始，step = 2, 取第0， 2，4列的值。

b[:, 1::2]表示取b的第0纬度的所有值，第一纬度从第1列开始，step = 2, 取第1， 3，5列的值。

[start :: step] 从start开始，每隔step取值。