Question

我可以轻松地（最好是一行）重塑一个列表

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

到

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

或

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

或

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

...

因为我传递了不同的参数（上面的例子为2,3,4）。

numpy.reshape不能这样做，因为列表的长度不能被2,3,4整除。

Answer 1

>>> l = [1,2,3,4,5,6,7,8,9,10,"prime"]
>>> [l[i:i+2] for i in xrange(0, len(l), 2)]
[[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], ['prime']]
>>> [l[i:i+3] for i in xrange(0, len(l), 3)]
[[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 'prime']]
>>> [l[i:i+4] for i in xrange(0, len(l), 4)]
[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 'prime']]

Answer 2

取自Python docs，适用于任何可迭代的答案：

在Python 3中：

from itertools import zip_longest

def grouper(n, iterable, fillvalue=None):
    "grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx"
    args = [iter(iterable)] * n
    return zip_longest(*args, fillvalue=fillvalue)

在Python 2.x中：

from itertools import izip_longest

def grouper(n, iterable, fillvalue=None):
    "grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx"
    args = [iter(iterable)] * n
    return izip_longest(fillvalue=fillvalue, *args)

Answer 3

或者您可以使用numpy.split：

data = np.array([1,2,3,4,5,6,7,8,9,10,"prime"], dtype=np.object)
np.split(data, np.r_[:len(data):3][1:])

使用不是列表长度的可分因子的形状进行整形

3 个答案: