Как перебирать список 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 и у меня проблемы. Есть идеи?
Пожалуйста, объясните, что именно означает «дополнительный последовательный элемент».
Быть новичком — это нормально, но все равно полезно поделиться своей попыткой, показать, что вы пробовали и что, по вашему мнению, может сработать, даже если это не сработало. Вы можете отредактировать вопрос, включив в него некоторый код.
Ответы 2
Вы можете использовать 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
Можно использовать более эффективные подходы, например бинарный поиск.
Также можно отбросить L и заменить его на range(1, N + 1), избегая ненужного временного list (и позволяя range выполнять всю простую математическую работу более эффективно, чем вручную добавлять единицу к каждому результату).
Вы уверены, что они хотят накопления? Насколько я понимаю, им нужен 1-й, 3-й, 6-й... элемент. Они останавливаются на 10, потому что 15-го элемента нет.
Да. @Лоцид прав. Он должен перебирать несколько элементов в исходном списке.
@CharlesJr Это список натуральных чисел или это просто список?
@Loocid прав, он должен работать с большими списками элементов из исходного списка данных. Я предполагаю, что могу изменить значение prefix_sum, чтобы включить все возможные элементы?
Судя по последнему комментарию:
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
По «дополнительному последовательному элементу». вы имеете в виду 1-й, 3-й, 6-й... элементы? Т.е. если
data = [9,5,3,1,5,10]ты хочешь9, 3, 10?