在python中保存生成的numpy随机数组而不使用列表?

问题描述:

我试图让情侣随机阵列与numpy.random.poisson。我想按照它们生成的顺序保存每个数据集。现在我正在使用列表来做这件事,我想知道是否有更好的方法来做到这一点,事后可以轻松访问每一组?在python中保存生成的numpy随机数组而不使用列表?

这是我的代码:

import numpy as np 
data = np.random.randint(0, 255, (250, 250, 3)) 
total = np.zeros(data.shape) 
listOfData = [] 
number = 10 #It could be any number 
for i in range(number): 
     new_total = np.random.poisson(lam=data) 
     listOfData.append(new_total.flatten()) 
     total += new_total 
firstSet = np.array(listOfData[0]).reshape(data.shape) 

我的数据是3 d阵列。

+0

你所说的“轻松访问每套”是什么意思? –

+0

我的意思是我可以在之后使用它们,只需从保存的地方调用它们的索引即可。就像说我可以使用第一个生成的数据集,只是调用ListOfData [0]之类的东西。 –

+0

你为什么不只是不'.flatten()'当你添加到列表中? –

在你的代码,listOfData[0]已经是一个ndarray,所以你可以跳过这一部分。