避免 Python 中的冗余循环

Question

我有两个列表（项目，销售），对于每对项目，两个列表之间的销售元素我必须调用一个函数。我正在寻找一种pythonic方法来避免这种冗余循环

第一个循环：

# Create item_sales_list

item_sales_list = list()

for item,sales in itertools.product(items,sales):
  if sales > 100:
    item_sales_list.append([item,sales])

result = some_func_1(item_sales_list)

第二次循环：

# Call a function with the result returned from first function (some_func_1)


for item,sales in itertools.product(items,sales):
   some_func_2(item,sales,result)

Answer 1

如果将结果存储在列表中，至少可以避免第二次调用itertools.product，在some_func_1的调用处添加条件：

item_sales_list = list(itertools.product(items, sales))

result = some_func_1([el for el in item_sales_list if el[1] > 100])

for item, sales in item_sales_list:
    some_func_2(item, sales, result)

除非您可以将不完整版本的result 传递给some_func_2，否则一次传递是不可能的。

Answer 2

一个解决方案，一个框架挑战。

首先，为了避免多次计算itertools.product()，您可以预先计算一次，然后将其用于两个循环：

item_product = list(itertools.product(items, sales))
item_sales_list = [[item, sales] for item, sales in item_product if sales > 100]

其次，循环两次实际上没有时间劣势（您仍然在做基本相同数量的工作 - 相同的操作，每次相同的次数。所以它仍然属于相同的复杂性等级）。在这种情况下，这是不可避免的，因为您需要第一次计算的结果（这需要遍历整个列表）来进行第二次计算。

result = some_func_1(item_sales_list)
for item, sales in item_product:
    some_func_2(item, sales, result)

如果您可以修改some_func_2() 使其不需要整个 item_sales_list 即可工作，那么您可以将其加载到相同的for 循环中并执行它们一个接一个。在不知道some_func_2() 的工作原理的情况下，不可能提供任何进一步的建议。