Как сделать эту функцию эффективной?

Question

[Игнатов Степан]

Есть функция:

def itogi(A):
    ANS = []
    for i in range(len(A)):
        ANS.append(sum(A[i]))
    return ANS

Она суммирует элементы в A[i] (Пример: ввод A=[[1,2,3],[4,5,0],[11,12,13]]) вывод ANS=[6,9,36]
Как можно сделать эту функцию эффективной?

Answer 1

[Сергей Горностаев]

def itogi(A):
    return [sum(i) for i in A]

Comments

[Игнатов Степан]

Будет быстрее чем та?

[Игнатов Степан]

Можно ли еще эффективнее

[Сергей Горностаев]

PupkinBro,

[Сергей Горностаев]

PupkinBro, быстрее уже некуда. Возможно, если данных будет очень много, удастся получить выигрыш в скорости используя NumPy:

import numpy as np

def itogi(A):
    return A.sum(axis=1)

itogi(np.array([[1,2,3],[4,5,0],[11,12,13]]))

Но на таких маленьких списках NumPy проигрывает в скорости.

[Игнатов Степан]

Сергей Горностаев, а есть ли алгоритм который быстрее посчитает сумму чем sum(A)?
к примеру a=[1,2,3,3,3,3,3,3,3,3,3,3,3]
я могу воспользоваться sum
но может есть алгоритм суммирования который быстрее?

[Сергей Горностаев]

PupkinBro, как думаете, хорошо ли создатели одного из самых знаменитых языков в мире разбираются в алгоритмах? Стали бы они использовать в sum() не самый эффективный алгоритм?

[Игнатов Степан]

Сергей Горностаев, Ну да согласен с Вами