对任意深度任意形式的list嵌套求平均

 2016.9.7  Post  181  1 分钟

def get_avg(L): 
     count=0
     def scantree(L): 
         nonlocal count 
         total = 0
         for x in L: 
             if not isinstance(x, list): 
                 total += x 
                 count += 1
             else: 
                 total += scantree(x) 
         return total 
     return scantree(L) / count 

 L =   [1, [2, [3, 4], 5], 6, [7, 8]]
 print(get_avg(L)) 

1
2
3
4
5
6
7
8
9

import collections


def get_item(iterable):
    for el in iterable:
        if not isinstance(el, str) and isinstance(el, collections.Iterable):
            yield from list_exp(el)
        else:
            yield el

1
2
3
4
5
6
7
8

L = [1, [2, [3, 4], 5], 6, [7, 8]]
sum, count = 0, 0
for i in get_item(L):
    count += 1
    sum += i
print(sum/count)

# 4.5