python - numpy - 获取随机整数作为 float

Question

所以在定点迭代中，我改变了矩阵初始化的方式

def init(M,N):
    return 2.5*np.ones([M,N])

到

def init(M,N):
    return nprnd.randint(1,6,[M,N])

在哪里

import numpy as np
import numpy.random as nprnd

这立即导致

A -= step_size * G

TypeError: Cannot cast ufunc subtract output from dtype('float64') to dtype('int32') with casting rule 'same_kind'

大不了，我就

nprnd.randint(1,6,[M,N], dtype='float')

除了

TypeError: Unsupported dtype "float64" for randint

我该如何避免这种情况？我有点不想生成一个矩阵然后复制它只是为了将它转换为 float 。有没有更智能的方式？

Answer 1

您可以使用 astype 进行显式转换:

nprnd.randint(1,6,[M,N]).astype("float")