fork download
  1.  
  2. from collections import defaultdict
  3.  
  4. a = ([0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2])
  5.  
  6. b = ([0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 4.3, 0.1, 0.2, 0.3, 0.4, 0.6] )
  7.  
  8. group_dict = defaultdict( list )
  9.  
  10.  
  11. # group every y by x
  12. for x, y in zip(a, b):
  13.  
  14. group_dict[x].append( y )
  15.  
  16.  
  17. # calculate average value of each group
  18. group_average = [ sum(group_dict[x])/len(group_dict[x]) for x in group_dict ]
  19.  
  20. # group id
  21. print( list( group_dict.keys() ) )
  22.  
  23. # group average
  24. print( group_average )
  25.  
Success #stdin #stdout 0.02s 9060KB
stdin
Standard input is empty
stdout
[0, 1, 2]
[0.3, 1.4599999999999997, 0.32]