[已解决]将列表拆分成指定个数的列表

gopythoner

RT:写一个函数foo(lis,n)传入2个参数，第一个是一个列表，第二个是需要拆分的个数
然后把指定的列表拆分成指定的个数（不需要每个列表的元素相同，毕竟这样也不一定能保证）
下面这个是我想的，但是感觉不简介

1. def foo(lis,n):
   
2.   if len(lis)%n == 0:
   
3.     new = [lis[i*len(lis)//n:(i+1)*len(lis)//n] for i in range(n)]
   
4.   else:
   
5.     new = [lis[i*len(lis)//n:(i+1)*len(lis)//n] for i in range(n-1)]
   
6.     new.append(lis[(n-1)*len(lis)//n:])
   
7.   print(new)
   
8.   return new
   
9.   
   
10. lis = list(range(23))
    
11. foo(lis,3)

复制代码

输出结果：[[0, 1, 2, 3, 4, 5, 6], [7, 8, 9, 10, 11, 12, 13, 14], [15, 16, 17, 18, 19, 20, 21, 22]]

有没有比较简介的？
lambda直接能搞定的那种？

最佳答案

月排行榜 / 总排行榜

小锟

2017-7-31 17:55:36

你可以看看numpy的split方法
https://docs.scipy.org/doc/numpy/reference/generated/numpy.split.html
另外一个array_split更符合
http://docs.scipy.org/doc/numpy/reference/generated/numpy.array_split.html

跳转到最佳答案楼层

小锟

你可以看看numpy的split方法
https://docs.scipy.org/doc/numpy/reference/generated/numpy.split.html
另外一个array_split更符合
http://docs.scipy.org/doc/numpy/reference/generated/numpy.array_split.html

SixPy

1. >>> def ls_split(ls,n):
   
2.         ln = len(ls)
   
3.         m = ln % n
   
4.         x = ln//n+(m>0)
   
5.         ret = list(zip(*([iter(ls)]*x)))
   
6.         if m>0 :ret.append(tuple(ls[ln//x*x:]))
   
7.         return  ret
   
8. 
   
9. >>> ls_split(range(21),3)
   
10. [(0, 1, 2, 3, 4, 5, 6), (7, 8, 9, 10, 11, 12, 13), (14, 15, 16, 17, 18, 19, 20)]
    
11. >>> ls_split(range(23),3)
    
12. [(0, 1, 2, 3, 4, 5, 6, 7), (8, 9, 10, 11, 12, 13, 14, 15), (16, 17, 18, 19, 20, 21, 22)]
    
13. >>>

复制代码

SixPy

1. >>> from itertools import zip_longest as zipl
   
2. >>> def ls_split(ls,n):
   
3.         return [list(filter(None,x))for x in zipl(*([iter(ls)]*(len(ls)//n+(len(ls)%n>0))))]
   
4. 
   
5. >>> ls_split(range(21),3)
   
6. [[1, 2, 3, 4, 5, 6], [7, 8, 9, 10, 11, 12, 13], [14, 15, 16, 17, 18, 19, 20]]
   
7. >>> ls_split(range(23),3)
   
8. [[1, 2, 3, 4, 5, 6, 7], [8, 9, 10, 11, 12, 13, 14, 15], [16, 17, 18, 19, 20, 21, 22]]
   
9. >>>

复制代码

Castieler

简单的来了

                               
登录/注册后可看大图

def foo(lis,n):
   for i in range(n):
      if i==n-1:
         print( lis[i:])
      else:
         print( lis[i:i+1])
foo([1,2,3,4,45,65],3)

冬雪雪冬

我写了一个，如果要写为labbda则需把s的计算重复几次写到一句中。

1. import math
   
2. def f(lst, n):
   
3.      s = math.ceil(len(lst) / n)
   
4.      return [lst[s * i: s * i + s] for i in range(n)]

复制代码

gopythoner

Castieler 发表于 2017-7-31 19:33
简单的来了

def foo(lis,n):

打印不是目的，return才是

Castieler

gopythoner 发表于 2017-8-1 19:24
打印不是目的，return才是

def foo( lis, n):
        a=[]
        for i in range(n):               
                if i == n - 1:                       
                        a.append(lis[i:])
                else:
                        a.append(lis[i:i + 1])
        return a
print(foo( [1, 2, 3, 4, 45, 65], 3))
改一下就行