在 Python 中快速操作嵌套循环

Question

我有 3 个类：Group、Customer、Product。每个组包含一个客户列表 [C1, C2, C3,...]，每个客户包含一个他想要购买的产品列表 [P1,P2,P3,...]。在顶层，我想做各种聚合，例如总订单值多少钱。随着组、客户和产品数量的增加，我最终会遇到非常慢的嵌套循环

total_order = 0
for customer in group:
    for product in customer:
        total_order += product.price * product.amount

在构建代码以使其更快的方面，您有什么建议？

Answer 1

这些团体有多大？如果这些操作是瓶颈，我会感到惊讶。您要在循环中使用数据库吗？

您可以改用列表推导式，但我不确定它能给您带来什么。

total_order = sum(p.price * p.amount for c in group for p in c)

Answer 2

在 python 中，您必须尽可能避免 for 循环，原因有很多。 即使在机器学习（神经网络）术语中，最好使用向量化而不是显式的 for 循环。 在您的情况下，我建议使用“map”，因为它实际上是用 C 编写的循环。