Как перебирать список Python, чтобы получить каждый дополнительный элемент списка?

У меня есть список Python, который мне нужно пройти, чтобы получить дополнительный последовательный элемент.

Я пытаюсь создать строку из каждого последующего элемента в моем списке с именем data.

data = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, ...]

Мне нужно превратить это в строку, которая читается как...

data_string = "1, 3, 6, 10"

Список data может содержать любое количество элементов и может меняться, но строка всегда должна состоять из элементов 1,3, 6, 10, 15, 21 и так далее.

Я попробовал запустить следующую функцию...

def create_lists(values_list):
    result = [[]]
    for value in values_list[1:]:
        new_lists = [curr_list + [value] for curr_list in result]
        result.extend(new_lists)
    return result

result = create_lists(values_list)
print(result)

При этом распечатываются только отдельные списки с отдельными элементами.

Я новичок в Python и у меня проблемы. Есть идеи?

🤔 А знаете ли вы, что...
Python поддерживает параллельное и асинхронное программирование с помощью модулей asyncio и multiprocessing.

[РЕШЕНО]Перейти к решению

102

Python303213 вопросов Loops11581 вопрос

Ответы:

Вы можете использовать accumulate() от itertools:

from itertools import accumulate

def _prefix_sum(N):
    L = [i + 1 for i in range(N)]
    psum = list(accumulate(L))
    return psum


print(_prefix_sum(13))

Если вы хотите иметь строковый вывод:

from itertools import accumulate


def _prefix_sum(N):
    L = [i + 1 for i in range(N)]
    return ', '.join(map(str, accumulate(L)))

print(_prefix_sum(13))

Принты

1, 3, 6, 10, 15, 21, 28, 36, 45, 55, 66, 78, 91

С таким же успехом можно отбросить L и заменить его на range(1, N + 1), избегая ненужного временного списка (и позволяя диапазону выполнять всю простую математическую работу более эффективно, чем вручную добавлять по одному к каждому результату). - Теневой Рейнджер

from itertools import accumulate

_prefix_sum = lambda N: ', '.join(map(str, accumulate(range(1, N + 1))))

print(_prefix_sum(13))

Редактировать

Чтобы найти сумму префикса в массиве, вы можете использовать set():

def _filter(L):
    L.sort()
    curr, res, psums = 0, '', set()
    full_sum = sum(L)
    for num in L:
        curr += num
        psums.add(curr)
        if num in psums:
            res += f"{num},"
        if curr > full_sum:
            break

    return res[:-1]


print(_filter([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]))
print(_filter([1, 2, 3, 7, 8, 9, 10, 50, 40, 20]))

Принты

1,3,6,10
1,3,40

Можно использовать более эффективные подходы, например бинарный поиск.

Решено

Судя по последнему комментарию:

data = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]

i, c, out = 0, 2, []
while i < len(data):
    out.append(str(data[i]))
    i += c
    c += 1

print(", ".join(out))

Распечатки:

1, 3, 6, 10

Как перебирать список Python, чтобы получить каждый дополнительный элемент списка?

Ответы:

Принты

Комментарии

Редактировать

Принты

Интересные вопросы для изучения