Как найти наименьшую сумму в питоне

Как рассчитать минимальную сумму элементов в матричном питоне? например:

z = 
np.array([[1, 2, 3], 

 [3, 2, 5], 

 [2, 4, 4]])

Минимальная сумма элемента: 1 + 2 + 2 = 5

Я устал:

def calculate(a):
    sum = 0
    for i in range(len(a)):
        sum += np.sum(a[i])

    return sum

Но это дает мне 26 вместо 5

Вы добавляете сумму всей строки к сумме, когда вы должны добавить минимальное значение. Так что просто замените sum на min:

z = np.array([[1, 2, 3], [3, 2, 5], [2, 4, 4]])

def calculate(a):
    sum = 0
    for row in a:
        sum += np.min(row)
    return sum

print(calculate(z))

>>> 5

Я также оптимизировал ваш метод цикла, чтобы быть более понятным.

Итак, вы z в качестве следующего массива. В NumPy вы должны понимать axis массива: (пожалуйста, посмотрите этот ответ для лучшего умственного образа: очень базовая-numpy-массив-измерение-визуализация). Итак, массив здесь 2D.

#----> axis: 1      #| a
array([[1, 2, 3],   #| x
       [3, 2, 5],   #| i
       [2, 4, 4]])  #| s
                    #v 0

Итак, мы находим минимум вдоль axis 1

In [70]: np.min(z, axis=1) 
Out[70]: array([1, 2, 2])

А так как вам также нужна сумма, вы вызываете .sum() для результирующего 1D массива из вышеприведенного шага. Ниже приведен код в одной строке:

In [71]: np.min(z, axis=1).sum()   
Out[71]: 5

Общее практическое правило заключается в том, чтобы не использовать циклы for с массивами NumPy, поскольку это будет очень медленно (с точки зрения времени выполнения).

Вы можете использовать np.min(): —

z = np.array([[1,2,3],[3,2,5],[2,4,4]])
res = sum(np.min(z, axis=1))
print(res)

Выход:

5

Вы не должны использовать циклы for для выполнения операций с массивами Numpy. Numpy существует, чтобы предоставлять оптимизированные альтернативы циклам, и использование циклов побеждает эту цель. Вы можете взять минимальное значение каждой строки (сворачивая первую ось), а затем суммировать получающиеся три значения:

np.sum(np.min(z, axis=1))

Out[61]: 5

Вы можете применить функцию, чтобы найти минимальное значение для каждой строки и сумму после этого.

import numpy as np
z = np.array([[1, 2, 3], 
             [3, 2, 5], 
             [2, 4, 4]])

np.apply_along_axis(np.min,-1,z).sum()

Как это работает? Сначала вы определяете функцию. Нахождение минимального значения, в этом случае. Во-вторых, вы определяете ось, которую вы хотите найти вам значения. Смотрите полное описание в документации ниже. Ласли я запускаю .sum (), чтобы суммировать все элементы. Для учебных целей я предлагаю только бег

np.apply_along_axis(np.min,-1,z)

Подробнее читайте в документации Numpy

Вы найдете np.min над осью 1.

z = np.array([[1, 2, 3], [3, 2, 5], [2, 4, 4]])
z.min(1).sum()
# 5

Пытаться:

min_sum = sum(map(np.min,z)) 
print(min_sum) # output: 5