Как создать массив массивов с динамическим размером с NumPy?

Question

Я хотел бы получить массив массивов numpy с диапазоном длины, например:

>>> source = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
>>> get_array_of_arrays_with_min_length(source, 5)
    array([[1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
           [2, 3, 4, 5, 6, 7, 8, 9, 10],
           [3, 4, 5, 6, 7, 8, 9, 10],
           [4, 5, 6, 7, 8, 9, 10],
           [5, 6, 7, 8, 9, 10],
           [6, 7, 8, 9, 10]])

Как это сделать с меньшим количеством кода?

Answer 1

Создайте свои списки,

l = [ [i+1 for i in range(x+1, 10)] for x in range(6)]

Затем создайте свой массив.

a = numpy.array(l)

Если бы мне пришлось угадывать, что должна делать ваша функция,

def get_array_of_arrays(source, m):
    return [ [ i for i in source if i>x] for x in range(m+1)]

Это даст вам результат, который вы запросили из предоставленных материалов.На самом деле главная идея здесь - составить список списков.