Свернуть массив nD numy в массив 1D - PullRequest
2 голосов
/ 13 марта 2012

Я пытаюсь суммировать значения массива nD вдоль определенной оси, чтобы эффективно свернуть его в одномерный массив.Я просматривал документы, но не смог найти нужную функцию.Я попытаюсь объяснить мой вопрос лучше с помощью некоторого кода:

In [46]: g
Out[46]: 
array([[1, 2, 3],
       [1, 2, 3],
       [1, 2, 3],
       [1, 2, 3],
       [1, 2, 3]])

Вывод, который мне нужен:

array([5,10,15])

Мои фактические данные представляют собой файл размером 7 МБ, поэтому я не оченьхочу использовать цикл for.

Спасибо за помощь

1 Ответ

6 голосов
/ 13 марта 2012

Просто делаю

numpy.sum(g, axis=0)

должно работать.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...