Матрица N x N x 3, получить первый элемент внутреннего массива 3

У меня есть матрица numpy, которая представляет изображение, и я хочу эффективно получить массив первого элемента каждого цвета. Матрица имеет размер N x M x 3 и x 3, потому что это цвета. Я хочу суммировать все, например, красные. Пример следующий:

a = np.random.randint(1,10,(2,4,3))
 a
array([[[1, 8, 8],[9, 9, 9],[4, 7, 7],[7, 1, 5]],[[3, 2, 2],[8, 5, 4],[3, 3, 5],[2, 2, 7]]])
sum_r = 0
for i in a:
    for j in i:
         sum_r += j[0]
print(sum_r)
37
1 + 9 + 4 + 7 + 3 + 8 + 3 + 2
37

Мне нужно сделать это эффективно, потому что N и M очень велики.

— 28.04.2019 08:52

a[:,:,0].sum() или a.sum(axis=(0,1))

— 28.04.2019 09:18

Я люблю тебя, правда, спасибо, мой генетический алгоритм был очень медленным, спас мою добычу.

— 28.04.2019 09:31

python-3.x numpy

28.04.2019 08:45

Стоит ли изучать PHP в 2026-2027 годах?

Привет всем, сегодня я хочу высказать свои соображения по поводу вопроса, который я уже много раз получал в своем сообществе: "Стоит ли изучать PHP в...

Поведение ключевого слова "this" в стрелочной функции в сравнении с нормальной функцией

В JavaScript одним из самых запутанных понятий является поведение ключевого слова "this" в стрелочной и обычной функциях.

Приемы CSS-макетирования - floats и Flexbox

Здравствуйте, друзья-студенты! Готовы совершенствовать свои навыки веб-дизайна? Сегодня в нашем путешествии мы рассмотрим приемы CSS-верстки - в...

Тестирование функциональных ngrx-эффектов в Angular 16 с помощью Jest

В системе управления состояниями ngrx, совместимой с Angular 16, появились функциональные эффекты. Это здорово и делает код определенно легче для...

Концепция локализации и ее применение в приложениях React ⚡️

Локализация - это процесс адаптации приложения к различным языкам и культурным требованиям. Это позволяет пользователям получить опыт, соответствующий...

Пользовательский скаляр GraphQL

Листовые узлы системы типов GraphQL называются скалярами. Достигнув скалярного типа, невозможно спуститься дальше по иерархии типов. Скалярный тип...

538

Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

def sumColumn(m):
    return [sum(col) for col in zip(*m)]


column = 0
matrix = [[1 , 4, 5, 12], 
          [-5, 8, 9, 0],
          [-6, 7, 11, 19]]

print("Sum of the elements in column", sumColumn(matrix)[column])

Спасибо, но я скорее искал быстрое эффективное решение, такое как [:, 0] или что-то в этом роде.

— 28.04.2019 09:17

28.04.2019 09:15