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

时间:2015-08-25 18:55:55

标签: python numpy multidimensional-array

我有一个numpy数组Z,以便:

Z.shape

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

如何将Z转换为新的数组NewZ,例如:

NewZ.shape

#Out[2]:
    (138, 112)

1 个答案:

答案 0 :(得分:1)

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

假设我们有

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

以下是两个示例,均产生NewZ.shape (138, 112)

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