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 awesomeness;我确信这是一个非常适合的机制，我无法想到它。

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

Answer 1

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

>>> l = [1,2,3,4,5,6,7,8,9,10]
>>> n = 4
>>> [l[i::n] for i in xrange(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)

按列重新排列python列表到n个列表

4 个答案: