Normalize function

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)。