Início / coding / Perguntas / 77482602
Accepted
enzo
Asked: 2023-11-15 01:20:34 +0800 CST

Como criar uma lista fragmentada de listas com comprimentos quase iguais?

  • 772

Preciso dividir um número em 3 partes. A seguinte função ( source ) faz isso corretamente:

def parts(num: int, div: int) -> list[int]:
    """Split a number into equal parts."""
    return [num // div + (1 if x < num % div else 0) for x in range(div)]

Então

assert parts(8, 3) == [3, 3, 2]
assert parts(9, 3) == [3, 3, 3]
assert parts(10, 3) == [4, 3, 3]

Dessa lista, preciso de uma lista de listas contendo um intervalo de 0 a num + 3, de modo que cada sublista

  • em um índice par tem comprimento de 1
  • em um índice ímpar tem comprimento de n, nde onde é obtidoparts
assert expected(8, 3) == [[0], [1, 2, 3], [4], [5, 6, 7], [8], [9, 10]]
assert expected(9, 3) == [[0], [1, 2, 3], [4], [5, 6, 7], [8], [9, 10, 11]]
assert expected(10, 3) == [[0], [1, 2, 3, 4], [5], [6, 7, 8], [9], [10, 11, 12]]

Isto é o que eu tentei:

def actual(num: int, div: int) -> list[list[int]]:
    matrix: list[list[int]] = []
    for i, length in enumerate(parts(num, div)):
        base = i * (length + 1)
        matrix.append([base])
        matrix.append([base + j + 1 for j in range(length)])

    return matrix

No entanto, só funciona quando numé divisível por div. Como posso consertar isso?

Estas são as saídas da minha função:

assert actual(8, 3) == [[0], [1, 2, 3], [4], [5, 6, 7], [6], [7, 8]]
assert actual(9, 3) == [[0], [1, 2, 3], [4], [5, 6, 7], [8], [9, 10, 11]]
assert actual(10, 3) == [[0], [1, 2, 3, 4], [4], [5, 6, 7], [8], [9, 10, 11]]
python

2 respostas

  1. Em vez de usar base = i * (length + 1), comece com base = 0(antes do loop) e apenas aumente basequantos elementos você adicionou nessa iteração.

    def actual(num: int, div: int) -> list[list[int]]:
    base = 0
    matrix: list[list[int]] = []
    for i, length in enumerate(parts(num, div)):
        matrix.append([base])
        matrix.append([base + j + 1 for j in range(length)])
        base += 1+ length
    return matrix
    • 1
  2. Best Answer

    Parece que cada sublista contém números crescentes. Isto é o que itertools.countpode lhe dar.

    Adaptei sua função para usá-la assim:

    from itertools import count

def actual(num: int, div: int) -> list[list[int]]:
    matrix: list[list[int]] = []
    base = count(0)
    for length in parts(num, div):
        matrix.append([next(base)])
        matrix.append([next(base) for _ in range(length)])

    return matrix
    • 1

relate perguntas