将列表拆分成更小的列表
我正在寻找一种方法来轻松地将python列表分成两半。
所以,如果我有一个数组:
A = [0,1,2,3,4,5]
我将能够得到:
B = [0,1,2] C = [3,4,5]
A = [1,2,3,4,5,6] B = A[:len(A)/2] C = A[len(A)/2:]
如果你想要一个function:
def split_list(a_list): half = len(a_list)/2 return a_list[:half], a_list[half:] A = [1,2,3,4,5,6] B, C = split_list(A)
一个更通用的解决scheme(你可以指定你想要的部分数量,而不是分成两半):
编辑 :更新后处理奇怪的列表长度
EDIT2 :根据Brians资讯性评论再次更新post
def split_list(alist, wanted_parts=1): length = len(alist) return [ alist[i*length // wanted_parts: (i+1)*length // wanted_parts] for i in range(wanted_parts) ] A = [0,1,2,3,4,5,6,7,8,9] print split_list(A, wanted_parts=1) print split_list(A, wanted_parts=2) print split_list(A, wanted_parts=8)
f = lambda A, n=3: [A[i:i+n] for i in range(0, len(A), n)] f(A)
n
– 结果数组的预定义长度
def split(arr, size): arrs = [] while len(arr) > size: pice = arr[:size] arrs.append(pice) arr = arr[size:] arrs.append(arr) return arrs
testing:
x=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13] print(split(x, 5))
结果:
[[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11, 12, 13]]
B,C=A[:len(A)/2],A[len(A)/2:]
这是一个常见的解决scheme,将arr分成count部分
def split(arr, count): return [arr[i::count] for i in range(count)]
如果你不关心订单…
def split(list): return list[::2], list[1::2]
list[::2]
从第0个元素开始,获取列表中的第二个元素。
list[1::2]
从第一个元素开始获取列表中的第二个元素。
def splitter(A): B = A[0:len(A)//2] C = A[len(A)//2:] return (B,C)
我testing了,并且需要双斜杠来强制python3中的int分割。我原来的post是正确的,尽pipewysiwyg在Opera中出了问题,出于某种原因。
对于将数组分割成n
较小的数组的更广义的情况,有一个官方的Python receipe。
from itertools import izip_longest def grouper(n, iterable, fillvalue=None): "Collect data into fixed-length chunks or blocks" # grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx args = [iter(iterable)] * n return izip_longest(fillvalue=fillvalue, *args)
这个代码片段来自Python的itertools文档页面 。
使用列表切片 。 语法基本上是my_list[start_index:end_index]
>>> i = [0,1,2,3,4,5] >>> i[:3] # same as i[0:3] - grabs from first to third index (0->2) [0, 1, 2] >>> i[3:] # same as i[3:len(i)] - grabs from fourth index to end [3, 4, 5]
要获得列表的前半部分,请从第一个索引切换到len(i)/2
…
>>> i[:len(i)/2] [0, 1, 2]
..交换价值左右,以获得下半场:
>>> i[len(i)/2:] [3, 4, 5]
虽然上面的答案或多或less是正确的,但如果数组的大小不能被2整除,那么由于a / 2
,a是奇数,在Python 3.0中是浮点数,而在早期版本,如果您from __future__ import division
脚本的开始处的from __future__ import division
指定。 在任何情况下,为了达到你的代码的“向前”兼容性,最好使用整数除法,即a // 2
。
来自@ChristopheD的提示
def line_split(N, K=1): length = len(N) return [N[i*length/K:(i+1)*length/K] for i in range(K)] A = [0,1,2,3,4,5,6,7,8,9] print line_split(A,1) print line_split(A,2)
这与其他解决scheme类似,但速度稍快。
# Usage: split_half([1,2,3,4,5]) Result: ([1, 2], [3, 4, 5]) def split_half(a): half = len(a) >> 1 return a[:half], a[half:]
#for python 3 A = [0,1,2,3,4,5] l = len(A)/2 B = A[:int(l)] C = A[int(l):]