如何在 Python 3.5 中将整数平均拆分为长度为 n 的列表？

Question

我需要能够将一个整数平均拆分为一个长度为 n 的列表。

例如 n = 4，

12 -> [0, 0, 1, 2]
1234 -> [1, 2, 3, 4]
12345 -> [1, 2, 3, 45]
123456 -> [1, 2, 34, 56]
1234567 -> [1, 23, 45, 67]
12345678 -> 12, 34, 56, 78]
123456789 -> [12, 34, 56, 789]

我敢肯定，我给出的例子太多了，但这有助于理解重点。

我过去用来将项目拆分为列表的代码是：

def split(s, chunk_size):
    a = zip(*[s[i::chunk_size] for i in range(chunk_size)])
    return [''.join(t) for t in a]

但是这段代码只分成 n 块。 （来自 StackOverflow 的代码，我不记得确切的帖子了，抱歉）。

感谢您的帮助。

Answer 1

def split(seq, n):
    """Partitions `seq` into `n` successive slices of roughly equal size.

    The sizes of each yielded slice shall differ by at most one.
    """
    q, r = divmod(len(seq), n)
    start = 0

    for i in range(n):
        size = q + (i < r)
        yield seq[start : start + size]
        start += size

它的工作方式与this popular answer 基本相同，只是它将剩余部分均匀地分配到每个 n 个切片中。

不幸的是，这将分配到列表的头部。要根据您的情况对其进行调整，您需要将 size = q + (i < r) 更改为：

size = q + (i >= (n - r))

通过此更改，以下内容：

print(list(split('1234', 4)))
print(list(split('12345', 4)))
print(list(split('123456', 4)))
print(list(split('1234567', 4)))
print(list(split('12345678', 4)))
print(list(split('123456789', 4)))

打印：

['1', '2', '3', '4']
['1', '2', '3', '45']
['1', '2', '34', '56']
['1', '23', '45', '67']
['12', '34', '56', '78']
['12', '34', '56', '789']

Answer 2

Ahsan 答案的变体应正确分布到序列末尾的其余部分：

from itertools import accumulate

def split(s, parts):
    s = str(s)
    r, rem = divmod(len(s), parts)
    norem = parts - rem
    counts = [r]*norem + [r+1]*rem
    # Prepend 0 to counts only for accumulate to start from 0
    return [s[start:start+cnt] for start, cnt in zip(accumulate([0] + counts), counts)]

Answer 3

选择的答案有其优势，但我以更直接的方式做到了：

def chunk(s, n):
    return [s[i:i+n] for i in range(0, len(s), n)]

def split(s0, numberOfChunks):
    s = ('0' * max(0, numberOfChunks - len(s0))) + s0
    sizeOfChunk = len(s) / numberOfChunks
    numberOfSmallChunks = numberOfChunks - len(s) % numberOfChunks
    return chunk(s[:numberOfSmallChunks * sizeOfChunk], sizeOfChunk) +
           chunk(s[sizeOfChunk*numberOfSmallChunks:], sizeOfChunk+1)

Answer 4

抱歉 - 代码错误 它按照规范工作

def split(s, list_size):
    s = str(s)
    assert len(s) >= list_size, "assertion failed"
    r, rem = divmod(len(s), list_size)
    lst = []
    for i in range(list_size):
        lst.append(s[i*r:(i+1)*r])
    for j in range(rem):
        lst[-(j+1)] += s[list_size + j]
    return lst