Как разделить измерение в массиве numpy, если оно не делится одинаково?

Question

У меня есть массив numpy формы (584, 161). Я хочу разделить это так, чтобы оно было (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))