2D numpy数组列表中的最大值数组

Question

a = np.random.rand（4,2,2）

array（[[[0.10723991,0.20947733]，         [0.02998244,0.66482933]]，

   [[ 0.1707575 ,  0.34311093],
    [ 0.7368306 ,  0.83202149]],

   [[ 0.10195594,  0.81748548],
    [ 0.51577596,  0.84025512]],

   [[ 0.64234408,  0.82321103],
    [ 0.78017441,  0.29398195]]])

如何为每个元素获取最大值数组：

数组（[0.64234408,0.82321103]，       [0.78017441,0.84025512]）

Answer 1

您可以使用numpy max函数查找特定轴（或轴）上的最大元素。

对于您提供np.max(a, axis=0)的示例，您正在寻找。

等效地，您可以使用数组对象的max方法：a.max(axis=0)。