递归地对列表值执行函数并将每个结果存储在 Python 中

Question

我有一些丑陋的代码可以做我想做的事情，即对列表中的每个值执行一个函数“x // 3 - 2”，直到该值为 0（并每次存储结果以相加稍后），但我想修改它以提供数百个列表值，而不仅仅是这三个。

fuellist = []
modules = [14, 1969, 100756]
def getfuel(lst):
  newlist = [x // 3 - 2 for x in lst]
  for x in newlist:
    if x < 0:
      newlist.remove(x)
  return newlist


list1 = getfuel(modules)
list2 = getfuel(list1)
list3 = getfuel(list2)
list4 = getfuel(list3)
list5 = getfuel(list4)
list6 = getfuel(list5)
list7 = getfuel(list6)
list8 = getfuel(list7)
list9 = getfuel(list8)
list10 = getfuel(list9)

total = list1 + list2 + list3 + list4 + list5 + list6 + list7 + list8 + list9 + list10
print(sum(total))

输出是/应该是这个列表：51314

我不记得最好的方法了。我试过使用while循环，但它陷入了某种无限循环。我不确定“递归”是否适合我正在寻找的东西，我希望我能更好地解释它。感谢您的帮助。

Answer 1

如果列表为空，您可以使用 while 循环，它会被评估为 false，所以

modules = [14, 1969, 100756]
def getfuel(lst):
    result = 0
    while lst:
        lst = [x for x in [x // 3 - 2 for x in lst] if x > 0]
        result+=sum(lst)
    return result

print(getfuel(modules)) # returns the sum

Answer 2

在您发表评论后，我已经更新了我的答案。这对你来说很好。这是解决问题的一种非常简单易懂的方法。

fuellist = []
modules = [14, 1969, 100756]
def getfuel(lst):
  newlist = [x // 3 - 2 for x in lst]
  for x in newlist:
    if x < 0:
      newlist.remove(x)
  return newlist

l = [getfuel(modules)]
for i in range(10):
    l.append(getfuel(l[i]))
s = 0
for i in l:
    s += sum(i)
print(s)

当您删除一个元素时，它会从列表中删除，但 for loop 在初始运行时会获取它的终止值，因此即使您稍后在循环中更改列表内容，它仍会运行到您拥有的位置最初设置它，所以它最终会显示错误或进入无限循环。
希望它能解决你的问题:)