scipy.random.rand()和同一个包中的其他函数
都生成float64数组作为输出
(至少对于Mac OS上的python 2.7.3 64位,scipy版本0.12.0).
我想要的是一个相当大的(N千兆字节)随机初始化的float32矩阵.
是否有一种简单的方法可以直接生成一个,而不是分配双倍空间
对于float64然后转换为32位？
最佳答案
我会预先分配数组,然后按照Warren Weckesser在评论中推荐的那样随机复制float64s.
>>> bytes_per_float = np.float32(0).nbytes # ugly,I know
>>> np.frombuffer(random.bytes(10 * bytes_per_float),dtype=np.float32)
array([ -3.42894422e-23,-3.33389699e-01,-7.63695071e-26,7.02152836e-10,3.45816648e-18,2.80226597e-09,-9.34621269e-10,-9.75820352e+08,2.95705402e+20,2.57654391e+25],dtype=float32)
当然,这些不遵循任何好的分布,阵列可能包含NaN或Inf,并且由于对齐问题,代码实际上可能在某些非x86机器上崩溃.