在Python中均匀分配列表

Question

假设我在Python中有以下这个列表：

a = ['a','b','c','d','e','f','g','h','i','j']

我想把这个列表分配成这样：

['a','f']
['b','g']
['c','h']
['d','i']
['e','j']

如果我的列表长度不一样，怎么把多出来的元素放到一个单独的列表里呢？

我希望能够把原始列表的元素分成n个部分，按照上面提到的方式。

比如说如果n=3，那结果会是：

['a','d','g']
['b','e','h']
['c','f','i']

而多出来的元素则放在一个单独的列表中

['j']

Answer 1

虽然不是特别优雅，但我来试试：

In [5]: a = ['a','b','c','d','e','f','g','h','i','j']

In [6]: [[a[i], a[len(a)//2+i]] for i in range(len(a)//2)]
Out[6]: [['a', 'f'], ['b', 'g'], ['c', 'h'], ['d', 'i'], ['e', 'j']]

如果你对一组元组的列表满意，可以使用 zip()：

In [7]: zip(a[:len(a)//2], a[len(a)//2:])
Out[7]: [('a', 'f'), ('b', 'g'), ('c', 'h'), ('d', 'i'), ('e', 'j')]

要把这个转换成一个列表的列表：

In [8]: map(list, zip(a[:len(a)//2], a[len(a)//2:]))
Out[8]: [['a', 'f'], ['b', 'g'], ['c', 'h'], ['d', 'i'], ['e', 'j']]

Answer 2

你可以使用zip和列表推导式一起使用：

def distribute(seq):
    n = len(seq)//2  #Will work in both Python 2 and 3
    return [list(x) for x in zip(seq[:n], seq[n:])]

print distribute(['a','b','c','d','e','f','g','h','i','j'])
#[['a', 'f'], ['b', 'g'], ['c', 'h'], ['d', 'i'], ['e', 'j']]