堆叠两个不同形状的两个 numpy 阵列

Question

我有两个 NumPy arrays。

一个具有维度 (4,1)。

我有另一个 numpy 阵列，尺寸为 (240, 320)

现在，我想制作一个如下所示的 CSV 文件：

由于两个 numpy 数组大小不同，我该如何堆叠它们？ 有什么建议吗？

Answer 1

我认为你不能根据 numpy.stack 堆叠两个没有相同形状的arrays 。 但是，您可以将 arrays 顺序保存到 csv 文件以获得您寻求的结果。 以下代码必须解决问题。

import numpy as np

mat0 = np.random.randint(0, 10, (4, 1), dtype=int)
mat1 = np.random.randint(0, 10, (240, 320), dtype=int)

with open('foo.csv', 'w') as f:
    np.savetxt(f, mat0, delimiter=',')
    np.savetxt(f, mat1, delimiter=',')