归一化函数

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)