Python - 将多个二维 arrays 加入一个 3D 数组

Question

所以我有多个文件可以访问并被视为 2D arrays。我想做的是将所有这些 2D arrays 放入一个 3D 数组中。 例如，如果我有 10 个形状为 (100,100) 的文件，当我组合它们时，应该留下形状为 (10,100,100) 的 3D 数组。 我的以下尝试如下：

filenames = glob.glob('source')
preset = np.empty([100,100], dtype = 'int16')
for file in filenames:
    data = fits.open(file)[0].data
    np.vstack([preset,data]).reshape((10,100,100))

但我得到的是以下错误：

ValueError: cannot reshape array of size 20000 into shape (10,100,100)

Answer 1

您正在成对执行操作。 尝试对所有 arrays 一起执行此操作：

arrs = [fits.open(file)[0].data for file in filenames]
np.vstack(arrs).reshape((10,100,100))

或者更直接：

np.stack(arrs)