python - 预分配多个形状相同的 numpy 数组

Question

我正在计算相同大小的 numpy 数组(模型网格数据)的许多属性。我更熟悉 matlab 语法，您可以使用类似于以下的命令预分配多个相同大小的数组:

[array1,array2,array3] = deal(NaN(size(array0)));

目前，我的 python2.7.7 (numpy1.9.0) 语法看起来非常糟糕、重复，而且非常不符合 Python 风格:

import numpy as np
array1 = np.ma.zeros(np.shape(array0))
array2 = array1.copy()
array3 = array1.copy()

是否有更好的方法来进行数组预分配？我考虑过使用列表，但我真的不确定最好的、最Pythonic的方式来简洁明了地实现这一点。

Answer 1

array1,array2,array3,array4 = [np.ma.zeros(np.shape(array0)) for _ in range(4)]

也可以...