要想弄清楚三维数组的切片问题,首先要明白三维数组的维度,定义一个三维数组,将其维度打印输出:
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]],
])
print(b.shape)
输出结果:
(3, 3, 4)
第一个数表示该三维数组中二维数组的个数,
第二个和第三个参数表示二维数组的行和列,
即这个三维数组中有3个3行4列的二维数组
接下来以实际的输出结果来逐一解释对这三个维度的操作问题:
- 第0维度:
print("b[0, ::]")
print(b[0, ::], b[0, ::].shape)
print("b[1, ::]")
print(b[1, ::], b[1, ::].shape)
print("b[-1, ::]")
print(b[-1, ::], b[-1, ::].shape)
print("b[0:2, ::]")
print(b[0:2, ::], b[0:2, ::].shape)
输出结果分析:
b[0, ::]
[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]] (3, 4)
第一个维度表示三维数组中二维数组的个数,
b[0, ::]将三维数组中的第一个二维数组输出
b[1, ::]
[[13 14 15 16]
[17 18 19 20]
[21 22 23 24]] (3, 4)
b[1, ::]将三维数组中的第二个二维数组输出
b[-1, ::]
[[25 26 27 28]
[29 30 31 32]
[33 34 35 36]] (3, 4)
负数表示从后向前逆向操作,
b[-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)
当出现b[m:n, ::]这样的形式,表示对第0维度中的(m,n-1)号元素进行操作
b[0:2, ::]表示取出第一个和第二个二维数组
- 第1维度:
print("b[:, 0:]")
print(b[:, 0:], b[:, 0:].shape)
print("b[:, 1:]")
print(b[:, 1:], b[:, 1:].shape)
print("b[:, -1:]")
print(b[:, -1:], b[:, -1:].shape)
print("b[:, 0:2:]")
print(b[:, 0:2:], b[:, 0:2:].shape)
输出结果分析:
b[:, 0:]
[[[ 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)
b[:, 0:]表示将每一个二维数组从第0行开始,全部输出,得到一个完整的三维数组
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)
b[:, 1:]表示将每一个二维数组从第1行开始,全部输出
b[:, -1:]
[[[ 9 10 11 12]]
[[21 22 23 24]]
[[33 34 35 36]]] (3, 1, 4)
b[:, -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)
b[:, 0:2:]表示将每一个二维数组的第0行和第1行全部输出
- 第2维度:
print("b[::, 0]")
print(b[::, 0], b[::, 0].shape)
print("b[::, 1]")
print(b[::, 1], b[::, 1].shape)
print("b[::, -1]")
print(b[::, -1], b[::, -1].shape)
print("b[::, 0:2:]")
print(b[::, 0:2:], b[::, 0:2].shape)
输出结果分析:
b[::, 0]
[[ 1 2 3 4]
[13 14 15 16]
[25 26 27 28]] (3, 4)
b[::, 0]取每个二维数组的第0行
b[::, 1]
[[ 5 6 7 8]
[17 18 19 20]
[29 30 31 32]] (3, 4)
b[::, 1]取每个二维数组的第1行
b[::, -1]
[[ 9 10 11 12]
[21 22 23 24]
[33 34 35 36]] (3, 4)
b[::, -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)
b[::, 0:2:]取每个二维数组的第0行和第1行
- 注意:第1维度和第2维度的区别:
如:b[:, 0:]取的是整个三维数组(从每个二维数组的第0行开始至尾行),而b[::, 0]取的是三维数组中的部分(每个二维数组的第0行)