Если вы работаете с массивами и вам нужно немного посчитать, я определенно рекомендую вам numpy Numpy был сделан для этой цели. Еще один намек (на все, что я знаю): старайтесь избегать петель, где можете.
Причина: код читается более четко и, скорее всего, будет быстрее.
Вот пример того, что может сделать numpy:
In [1]: import numpy as np
In [2]: x = np.array([4,5,6,7,8])
In [3]: x+3
Out[3]: array([ 7, 8, 9, 10, 11])
In [4]: x**2
Out[4]: array([16, 25, 36, 49, 64])
In [5]: x>=6
Out[5]: array([False, False, True, True, True], dtype=bool)
Для дальнейшего чтения я рекомендую учебник numpy .