2010-01-28 17 views
11

Tengo una listaPython: Encontrar promedio de una lista anidada

a = [[1,2,3],[4,5,6],[7,8,9]] 

Ahora quiero encontrar el promedio de estas listas interior para que

a = [(1+4+7)/3,(2+5+8)/3,(3+6+9)/3] 

'a' no debería ser un anidado lista al final. Sírvase proporcionar una respuesta para el caso genérico

+0

lo probaste? – SilentGhost

Respuesta

4
>>> import itertools 
>>> [sum(x)/len(x) for x in itertools.izip(*a)] 
[4, 5, 6] 
10
a = [sum(x)/len(x) for x in zip(*a)] 
# a is now [4, 5, 6] for your example 

En Python 2.x, si no desea que la división entera, vuelva a colocar sum(x)/len(x) por 1.0*sum(x)/len(x) anteriormente.

Documentation for zip.

+0

+1 para zip, pero ¿podría ahorrarse la molestia de una función extra? ;) –

+0

Pensé que el OP no conocía demasiado a Python. Editando de todos modos :-) –

4

Si ha instalado numpy:

>>> import numpy as np 
>>> a = [[1,2,3],[4,5,6],[7,8,9]] 
>>> arr = np.array(a) 
>>> arr 
array([[1, 2, 3], 
     [4, 5, 6], 
     [7, 8, 9]]) 
>>> np.mean(arr) 
5.0 
>>> np.mean(arr,axis=0) 
array([ 4., 5., 6.]) 
>>> np.mean(arr,axis=1) 
array([ 2., 5., 8.]) 
Cuestiones relacionadas