В настоящее время я использую стандартный цикл for, но мне бы хотелось векторизуйте эту операцию для повышения производительности, используя только numpy.
Пример
Код: Выделить всё
import numpy as np
indices = np.array([np.array([1, 2, 3]), np.array([4, 5]), np.array([6, 7, 8, 9])], dtype=object)
group_lengths = np.array([len(group) for group in indices])
print(group_lengths)
Код: Выделить всё
[3 2 4]Обновить
Спасибо всем за ваши идеи и разъяснения! Теперь я понимаю, что полная векторизация моего варианта использования невозможна. Я буду придерживаться своего текущего решения, поскольку оно кажется наиболее эффективным в данных обстоятельствах.
Подробнее здесь: https://stackoverflow.com/questions/790 ... torization
Мобильная версия