如果 numpy 数组中的维度不能相等地整除，我该如何拆分？

Question

我有一个(584, 161)形状的numpy数组。我想把它拆分成(6, 100, 161)，但在最后一行(？(，用0右键填充元素。

因此，如果我有(1412, 161)，它应该转换为(15, 100, 161)，在右侧的最后一行有88个0

Answer 1

你可以这样做：

import numpy as np
arr = np.random.rand(584, 161)
arr = np.append(arr, np.zeros((16, 161)), axis=0)
arr = arr.reshape((6, 100, 161))

或

arr = np.random.rand(584, 161)
arr_zero = np.zeros((16, 161))
arr = np.concatenate((arr, arr_zero), axis=0)
arr = arr.reshape((6, 100, 161))