多个列表从中循环取数

lovesword

例如 有a,b,c 3个列表，以及指定的要取数的个数n,依次从a,b,c三个列表取元素，凑足n个数 例如：

1. 
   
2. n=3
   
3. a=[1,2,3,4,5,6]
   
4. b=[7,8,9]
   
5. c=[10]
   
6. 
   
7. ----希望得到----
   
8. n=3  ===> [1,7,10]
   
9. n=4  ===> [1,7,10,2]
   
10. n=5  ===> [1,7,10,2,8]
    
11. n=6  ===> [1,7,10,2,8,3]
    

复制代码

graceasyi

帮你写了个，不知道有没有更好的方法。

1. def fun(a, b, c, n):
   
2.     new = []
   
3.     for i in range(max([len(a), len(b), len(c)])):
   
4.         try:
   
5.             new.append(a[i])
   
6.         except IndexError:
   
7.             pass
   
8.         try:
   
9.             new.append(b[i])
   
10.         except IndexError:
    
11.             pass
    
12.         try:
    
13.             new.append(c[i])
    
14.         except IndexError:
    
15.             pass         
    
16.     return new[:n]
    
17.                        
    
18.    
    
19. n=5
    
20. a=[1, 2, 3, 4, 5, 6]
    
21. b=[7, 8, 9]
    
22. c=[10]
    
23. print(fun(a, b, c, n))      
    

复制代码

colinshi

1. def xxx(n,*args):
   
2.     tmp = []
   
3.     count = n
   
4.     while count:
   
5.         for i in range(len(args)):
   
6.             if args[i]:
   
7.                 tmp.append(args[i].pop(0))
   
8.         count -= 1
   
9.         if len(tmp) >= n:
   
10.             break
    
11.     return tmp[:n]
    
12. if __name__ == '__main__':
    
13.     ab=xxx(n,a,b,c)
    
14.     print(ab)
    

复制代码

看一下结果吧

冬雪雪冬

也写一个：

1. a=[1,2,3,4,5,6]
   
2. b=[7,8,9]
   
3. c=[10]
   
4. 
   
5. def fun(n, *args):
   
6.     import copy
   
7.     list1 = copy.deepcopy(args)
   
8.     newlist = []
   
9.     while len(newlist) < n:
   
10.         for i in range(len(list1)):
    
11.             if list1[i]:
    
12.                 newlist.append(list1[i].pop(0))
    
13.                 if len(newlist) >= n:
    
14.                     break
    
15.     return newlist

复制代码