numpy获取随机整数作为浮点数

Question

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

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

到

^{pr2}$

在哪里

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