python - 在 numpy 中缩放(或规范化)这样的数组？

Question

在 numpy 中，原始数组的形状是这样的 (2,2,2)

[[[0.2,0.3],[0.1,0.5]],[[0.1,0.3],[0.1,0.4]]]

我想缩放数组，使 a 维度的最大值为 1，如下所示:

由于max([0.2,0.1,0.1,0.1])是0.2，1/0.2是5，所以对于int元组的第一个元素，乘以5。

由于max([0.3,0.5,0.3,0.4])为0.5，1/0.5为2，所以对于int元组的第二个元素，乘以2

所以最后的数组是这样的:

[[[1,0.6],[0.5,1]],[[0.5,0.6],[0.5,0.8]]]

我知道如何在 numpy 中将一个数组与一个整数相乘，但我不确定如何将数组与不同的因子相乘。有人对此有想法吗？

Answer 1

如果您的数组 = a:

>>> import numpy as np
>>> a = np.array([[[0.2,0.3],[0.1,0.5]],[[0.1,0.3],[0.1,0.4]]])

你可以这样做:

>>> a/np.amax(a.reshape(4,2),axis=0)
array([[[ 1. ,  0.6],
        [ 0.5,  1. ]],

       [[ 0.5,  0.6],
        [ 0.5,  0.8]]])