Numpy массив массива в массив длины (векторизация)Python

Программы на Python
Ответить
Anonymous
 Numpy массив массива в массив длины (векторизация)

Сообщение Anonymous »

У меня есть массив массивов, где каждый подмассив представляет группу элементов. Я хочу вычислить длину каждого подмассива и сохранить результат как новый массив numpy.
В настоящее время я использую стандартный цикл 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)
Ожидаемый результат Я ищу полностью векторизованное решение, позволяющее избежать явного цикла. Как этого можно достичь с помощью numpy?
Обновить
Спасибо всем за ваши идеи и разъяснения! Теперь я понимаю, что полная векторизация моего варианта использования невозможна. Я буду придерживаться своего текущего решения, поскольку оно кажется наиболее эффективным в данных обстоятельствах.

Подробнее здесь: https://stackoverflow.com/questions/790 ... torization
Ответить

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

Вернуться в «Python»