python分片

python分片笔记

import numpy as np

outputs = np.array([
    [
        [1, 2, 3],
        [4, 5, 6]
    ],
    [
        [7, 8, 9],
        [10, 11, 12]
    ],
    [
        [13, 14, 15],
        [16, 17, 18]
    ]
])

# 一维情况
# print(outputs[0])             # 获取第一个元素
# print(outputs[0][:,0])        # 返回outputs[0]中每个一维数组的第一个元素
# print(outputs[-2:-1])         # 获取倒数第二个元素
# print(outputs[::-1])          # 倒序
# print(outputs[-2:])             # a[-2:]表示从倒数第二个元素到序列末尾的所有元素。

# 步长
#print(outputs[::-2])            # 步长也可以是负数，表示倒序截取序列

# 三维分片
# print(outputs[:2,:,:])          # 提取前两个二维数组
# print(outputs[:,0,:])           # 提取每个二维数组的第一行
print(outputs[0][0,:])