numpy - 将随机整数作为浮点数

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：无法使用强制转换规则＆＃39; same_kind＆＃39;

将ufunc从dtype（＆＃39; float64＆＃39;）的输出减去dtype（＆＃39; int32＆＃39;）

很重要，我只是

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

除

  

TypeError：不支持的dtype＆＃34; float64＆＃34;对于randint

我如何规避这个？我有点不想生成一个矩阵，然后复制它只是为了将其转换为浮动。有更智能的方式吗？

Answer 1

您可以使用astype进行显式演员：

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