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

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 文档，适用于任何可迭代的答案：

在 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:])