Numpy数组切片总结

布满荆棘的人生 2022-12-25 00:59 156阅读 0赞

数组切片规则：\[*start*：*end*\]

还可以定义步长：\[*start*：*end*：*step*\]

若不传递 *start*，则将其视为 0。例如 \[:3\] 等价于 \[0:3\]

若不传递 *end*，则视为该维度内数组的长度。例如\[1:\] 等价于从第一个元素开始一直截取到末尾。

若不传递 *step*，则视为 1,即默认步长为1,可以省略不写。

特别的：\[::2\]  表示隔一个元素取一个,因为 start 没写默认为0, end 也没写默认为截取到末尾，步长设定为2。

**注:元素下标从0开始**

**一些例子:**

# -*- coding: utf-8 -*-
    import numpy as np
    
    b = np.array([[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]],
                  [[13, 14, 15, 16], [17, 18, 19, 20], [21, 22, 23, 24]],
                  [[25, 26, 27, 28], [29, 30, 31, 32], [33, 34, 35, 36]],
                  ])
    
    
    #逗号分隔维度,维度下标从0开始
    #数组元素下标也是从0开始
    print("原始数组:")
    print(b)
    print("数组类型: ",b.shape)
    print()
    
    print("---------------------------------------")
    print("第0维中第0个元素全选")
    print("b[0, ::]为:")
    print(b[0, ::])             #第0维中第0个元素全选
    print("数组类型:", b[0, ::].shape)
    print("---------------------------------------")
    print()
    
    print("第0维中第0到1个元素全选")
    print("b[0:2, ::]为:")
    print(b[0:2, ::])         #第0维中第0到1个元素全选
    print("数组类型:", b[0:2, ::].shape)
    print("---------------------------------------")
    print()
    
    print("第1维中从第1个元素开始往后全选")
    print("b[:, 1:]为:")
    print(b[:, 1:])             #第1维中从第1个元素开始往后全选
    print("数组类型:", b[:, 1:].shape)
    print("---------------------------------------")
    print()
    
    print("---------------------------------------")
    print("第1维中第0、1个元素全选")
    print("b[:, 0:2:]为:")
    print(b[:, 0:2:])             #第1维中第0、1个元素全选
    print("数组类型:", b[:, 0:2:].shape)
    print("---------------------------------------")
    print()
    
    print("---------------------------------------")
    print()
    print("第1维中第0个元素全选")
    print("b[::, 0]为:")
    print(b[::, 0])             #第1维中第0个元素全选
    print("数组类型:", b[::, 0].shape)
    print("---------------------------------------")
    print()
    
    
    print("---------------------------------------")
    print("第2维中第0个元素全选")
    print("b[:, :, 0]为:")
    print(b[:, :, 0])             #第2维中第0个元素全选
    print("数组类型:", b[:, :, 0].shape)
    print("---------------------------------------")
    print()

** 代码运行结果:**

原始数组:
    [[[ 1  2  3  4]
      [ 5  6  7  8]
      [ 9 10 11 12]]
    
     [[13 14 15 16]
      [17 18 19 20]
      [21 22 23 24]]
    
     [[25 26 27 28]
      [29 30 31 32]
      [33 34 35 36]]]
    数组类型:  (3, 3, 4)
    
    ---------------------------------------
    第0维中第0个元素全选
    b[0, ::]为:
    [[ 1  2  3  4]
     [ 5  6  7  8]
     [ 9 10 11 12]]
    数组类型: (3, 4)
    ---------------------------------------
    
    第0维中第0到1个元素全选
    b[0:2, ::]为:
    [[[ 1  2  3  4]
      [ 5  6  7  8]
      [ 9 10 11 12]]
    
     [[13 14 15 16]
      [17 18 19 20]
      [21 22 23 24]]]
    数组类型: (2, 3, 4)
    ---------------------------------------
    
    第1维中从第1个元素开始往后全选
    b[:, 1:]为:
    [[[ 5  6  7  8]
      [ 9 10 11 12]]
    
     [[17 18 19 20]
      [21 22 23 24]]
    
     [[29 30 31 32]
      [33 34 35 36]]]
    数组类型: (3, 2, 4)
    ---------------------------------------
    
    ---------------------------------------
    第1维中第0、1个元素全选
    b[:, 0:2:]为:
    [[[ 1  2  3  4]
      [ 5  6  7  8]]
    
     [[13 14 15 16]
      [17 18 19 20]]
    
     [[25 26 27 28]
      [29 30 31 32]]]
    数组类型: (3, 2, 4)
    ---------------------------------------
    
    ---------------------------------------
    
    第1维中第0个元素全选
    b[::, 0]为:
    [[ 1  2  3  4]
     [13 14 15 16]
     [25 26 27 28]]
    数组类型: (3, 4)
    ---------------------------------------
    
    ---------------------------------------
    第2维中第0个元素全选
    b[:, :, 0]为:
    [[ 1  5  9]
     [13 17 21]
     [25 29 33]]
    数组类型: (3, 3)
    ---------------------------------------