如何将列表分解为一对列表

Question

假设我有一个size=(8,64,1,60,60)列表，并希望将其分解为(4,2,64,1,60,60)，然后沿轴1求和。我尝试了下面的代码，但它抛出了错误:

'list' object has no attribute 'reshape'.

请注意，我想把预测作为一个列表，而不想把它改成数组。

predictions=list(np.random.randint(5,size=(8,64,1,60,60)))
predictions_sum = predictions.reshape(4,2, *predictions.shape[1:]).sum(axis=1)

Answer 1

你混淆了Python的内置list类型与numpy的array。试试这个:

predictions=np.array(np.random.randint(5,size=(8,64,1,60,60)))
predictions_sum = predictions.reshape(4,2, *predictions.shape[1:]).sum(axis=1)