规范化函数

Question

为了正常化，这行得通吗

def normalize(arr, axis):
    x = [i for i in range(len(arr.shape))]
    a = x[0]
    x[0] = x[axis]
    x[axis] = a
    arr = arr.transpose(x)
    return (arr/np.sum(arr**2, axis=0)**0.5).transpose(tuple(x))

Answer 1

如果你想正常化，你可以做x = x / sum(x)。这适用于numpy阵列：

import numpy

x = [1, 2, 3]
x = numpy.array(x)
x = x / sum(x)