python - 按列将 python 列表重新排列为 n 个列表

Question

我想将列表重新排列为列表l列表n，其中n是列数。

例如，

l = [1,2,3,4,5,6,7,8,9,10]
n = 4
==> [[1,5,9],[2,6,10],[3,7][4,8]]

有人可以帮我解决算法吗？随意使用任何可用的 python 真棒；我确信有一些很酷的机制非常适合这个，我只是想不出它。

PS 示例列表恰好是从 1 开始的有序数字。这不是我的实际情况。

Answer 1

在 Python 中确实有一个很酷的机制：切片的三参数形式，其中最后一个参数是步长。

>>> l = [1,2,3,4,5,6,7,8,9,10]
>>> n = 4
>>> [l[i::n] for i in range(n)]
[[1, 5, 9], [2, 6, 10], [3, 7], [4, 8]]

Answer 2

以下功能可以实现您想要实现的目标：

def rearrange(seq,n):
    return [[v for i,v in enumerate(seq[x:]) if i%n==0] for x in xrange(len(seq))][:n]

Answer 3

l = [1,2,3,4,5,6,7,8,9,10]
n = 4

def f(l,n):
    A = []
    [A.append([]) for i in xrange(n)] 
    [ A [(i - 1) % n].append(i) for i in l]
    return A

print f(l,n)

[[1, 5, 9], [2, 6, 10], [3, 7], [4, 8]]

Answer 4

编写 Python 不是打代码高尔夫的游戏，为了可读性，不要害怕使用多行代码。

l = [1,2,3,4,5,6,7,8]

def split_into_columns(input_list, num_of_cols=3):
    retval = [ [] for _ in xrange(num_of_cols)]      # build 3 columns
    for i in xrange(len(input_list)):                # iterate through original list
        retval[i%num_of_cols].append(input_list[i])  # place in the "modulo 3" column
    return retval

    # here's a compressed, less readable version of that for-loop
    #[retval[i%3].append(input_list[i]) for i in xrange(len(input_list))]
    #return retval

print split_into_columns(l, 3)