Быстрый способ расширить список в список индексов ⇐ Python

[Anonymous]

Учитывая индексный список разделения t длины M + 1 , где первый элемент - 0 , а последний элемент - n , генерируйте массив D длины n , что d [t : t [i + 1]] = I .
, например, date t = n, 2, 2, 2, 2, 7, 7, 7
, например, t = dable t = 2, 2, 2, 2, 7, 7
. D = [0, 0, 1, 1, 1, 2, 2] .
Я стараюсь избежать цикла, но лучшее, что я могу сделать:

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

def expand_split_list(split_list):
return np.concatenate(
[
np.full(split_list[i + 1] - split_list[i], i)
for i in range(len(split_list) - 1)
]
)

Есть ли встроенная функция для этой цели?


Подробнее здесь: https://stackoverflow.com/questions/795 ... index-list