element-wise addition on two lists

benchmark.py

#!/usr/bin/env python3

from timeit import timeit

NUMBER = 1000
ELEMENTS = 10000
SETUP = 'N = {}; X = list(range(N)); Y = list(range(N))'.format(ELEMENTS)
SETUP_NP = 'N = {}; import numpy as np; X = np.arange(N); Y = np.arange(N)'.format(ELEMENTS)

STATS = [
    ['map with operator.add',
     'list(map(add, X, Y))', SETUP + '; from operator import add'],
    ['map with lambda',
     'list(map(add, X, Y))', SETUP + '; add = lambda x, y: x + y'],
    ['map with sum and zip',
     'list(map(sum, zip(X, Y)))', SETUP],
    ['list comprehension with sum and zip',
     '[sum(z) for z in zip(X, Y)]', SETUP],
    ['list comprehension with + and zip',
     '[x + y for x, y in zip(X, Y)]', SETUP],
    ['numpy',
     'X + Y', SETUP_NP],    
]

NAME_WIDTH = max(len(stat[0]) for stat in STATS)
STAT_WIDTH = max(len(stat[1]) for stat in STATS)

print('With {:,} elements, average time of {:,} runs'.format(ELEMENTS, NUMBER))
for stat in STATS:
    t = timeit(*stat[1:], number=NUMBER)
    print('{:{name_width}}: {:{stat_width}} = {:7,.0f} us'.format(
          stat[0], stat[1], t / NUMBER * 1000000,
          name_width=NAME_WIDTH, stat_width=STAT_WIDTH))

elemadd.py

#!/usr/bin/env python3

X = range(4)
Y = range(4)

from operator import add

OP_ADD = list(map(add, X, Y))
LAMBDA = list(map(lambda x, y: x + y, X, Y))
SUMZIP = list(map(sum, zip(X, Y)))
COMSUM = [sum(z) for z in zip(X, Y)]
COMZIP = [x + y for x, y in zip(X, Y)]

import numpy as np

AX = np.arange(4)
AY = np.arange(4)
NP_ADD = AX + AY