Python numpy 3D数组到2D数组(移除维度)

Question

我有一个numpy数组Z，这样:

Z.shape
#Out[1]:
    (138, 112, 123)

如何将Z转换为新的数组NewZ，从而:

NewZ.shape
#Out[2]:
    (138, 112)

?

Answer 1

删除一个维度意味着删除信息，因此您必须决定将原始数据投影到较低维度数量的规则。

假设我们有

import numpy as np
Z = np.random.random((138, 112, 123))

这里有两个例子，都产生(138, 112)的NewZ.shape:

• NewZ = np.max(Z, axis=2)，取最后一个轴的最大元素。
• NewZ = Z[:,:,0]，取最后一个轴的第一个元素。