Is there any pythonic way to find average of specific tuple elements in array? Is there any pythonic way to find average of specific tuple elements in array? python python

Is there any pythonic way to find average of specific tuple elements in array?


If you are using Python 3.4 or above, you could use the statistics module:

from statistics import meanaverage = mean(value[1] for value in array)

Or if you're using a version of Python older than 3.4:

average = sum(value[1] for value in array) / len(array)

These solutions both use a nice feature of Python called a generator expression. The loop

value[1] for value in array

creates a new sequence in a timely and memory efficient manner. See PEP 289 -- Generator Expressions.

If you're using Python 2, and you're summing integers, we will have integer division, which will truncate the result, e.g:

>>> 25 / 46>>> 25 / float(4)6.25

To ensure we don't have integer division we could set the starting value of sum to be the float value 0.0. However, this also means we have to make the generator expression explicit with parentheses, otherwise it's a syntax error, and it's less pretty, as noted in the comments:

average = sum((value[1] for value in array), 0.0) / len(array)

It's probably best to use fsum from the math module which will return a float:

from math import fsumaverage = fsum(value[1] for value in array) / len(array)


If you do want to use numpy, cast it to a numpy.array and select the axis you want using numpy indexing:

import numpy as nparray = np.array([('a', 5) , ('b', 10), ('c', 20), ('d', 3), ('e', 2)])print(array[:,1].astype(float).mean())# 8.0

The cast to a numeric type is needed because the original array contains both strings and numbers and is therefore of type object. In this case you could use float or int, it makes no difference.


If you're open to more golf-like solutions, you can transpose your array with vanilla python, get a list of just the numbers, and calculate the mean with

sum(zip(*array)[1])/len(array)