python - 规范化 numpy 数组

Question

给定一个数组，我想对其进行归一化，使每一行总和为 1。

我目前有以下代码:

import numpy
w = numpy.array([[0, 1, 0, 1, 0, 0], 
                 [1, 0, 0, 0, 0, 1], 
                 [0, 0, 0, 0, 0, 1], 
                 [1, 0, 0, 0, 1, 0], 
                 [0, 0, 0, 1, 0, 1], 
                 [0, 1, 1, 0, 1, 0]], dtype = float)


def rownormalize(array):
    i = 0
    for row in array:
        array[i,:] = array[i,:]/sum(row)
        i += 1

我有两个问题:

1) 代码有效，但我想知道是否有更优雅的方法。

2) 如果数据类型是 int，如何将其转换为 float？我试过了

if array.dtype == int:
    array.dtype = float

但它不起作用。

Answer 1

你可以 1) 那样做:

array /= array.sum(axis=1, keepdims=True)

和 2) 像那样:

array = array.astype(float)