Как создать 3D-массив Numpy разной длины во 2-м измерении? ⇐ Python

[Anonymous]

У меня есть код, который создает трехмерный массив (

Код: Выделить всё

x_3d

) «на лету», используя значения из 2D-массива (

Код: Выделить всё

x

) в цикле for:

Код: Выделить всё

x_3d = np.empty((0, 20, 10))
for i in range(num_samples):
x_3d = np.append(x_3d, [x[i*20:(i+1)*20, :]], axis=0)

Результирующая форма 3D-массива равна (num_samples, 20, 10).
Если я хочу взять фрагменты разных длины из 2D-массива, чтобы количество строк менялось, как это сделать? Я рассмотрел Как индексировать несколько 2D-массивов с разным количеством строк в 3D-массиве?
Сначала сохраняя 2D-массивы в списке, а затем преобразуя список обратно в массив , я получил форму (num_samples,). Хотя каждый элемент представляет собой 2D-массив numpy, он не является 3D-массивом numpy формы (num_samples, length_varies, 10).

Подробнее здесь: https://stackoverflow.com/questions/536 ... -dimension