2010-01-28 4 views
11

J'ai une listePython: Trouver moyen d'une liste imbriquée

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

Maintenant, je veux trouver la moyenne de ces liste intérieure afin que

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

« a » ne devrait pas être imbriquées liste à la fin. Veuillez fournir une réponse pour le cas générique

+0

qu'avez-vous essayé? – SilentGhost

Répondre

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 vous ne voulez pas la division entière, remplacez sum(x)/len(x) par 1.0*sum(x)/len(x) ci-dessus.

Documentation for zip.

+0

+1 pour le zip, mais ne pourriez-vous vous épargner le problème d'une fonction supplémentaire? ;) –

+0

J'ai pensé que l'OP ne connaissait pas trop bien Python. Editer quand même :-) –

4

Si vous avez numpy installé:

>>> 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.]) 
Questions connexes