列表理解：带条件的双循环

Question

我尝试在 Python 3 中的列表理解方面训练自己，但我遇到了一个问题。 我在其他主题中看到了双循环语法，但在第二个循环中从来没有条件。

所以，我不得不说，在样本值和最大容量中，我能不能给一个样本充电。

我有一本字典 samples，其中包含样本的 id 及其 A/B/C 值。 我有一个 dict maximum 我可以充电的最大容量。 还有一个 samples_to_check 列表，其中包含我必须返回答案的样本。

samples_to_check = [0, 1, 2]

samples = {0: {"A": 2, "B": 0, "C": 3},
           1: {"A": 1, "B": 1, "C": 2},
           2: {"A": 0, "B": 0, "C": 5},
           3: {"A": 0, "B": 0, "C": 2}}

maximum = {"A": 2, "B": 2, "C": 2}

在这个例子中，我们只能对样本 1 收费，因为 A、B 和 C

result2 = [
    "CHARGE " + str(sample_id) for sample_id in samples_to_check
    if maximum["A"] - samples[sample_id]["A"] >= 0 and maximum["B"] - samples[sample_id]["B"] >= 0 and maximum["C"] - samples[sample_id]["C"] >= 0
]

print(result2)
['CHARGE 1']

但我想知道是否可以在不为每个字母重复相同条件的情况下找到最佳解决方案。

我试过这个，但它不起作用。

result = [
    "CHARGE " + str(sample_id) for sample_id in samples_to_check
    for letter in ["A", "B", "C"] if maximum[letter] - samples[sample_id][letter] >= 0
]

print(result)
['CHARGE 0', 'CHARGE 0', 'CHARGE 1', 'CHARGE 1', 'CHARGE 1', 'CHARGE 2', 'CHARGE 2']

谢谢。

Answer 1

您可以使用 all/any 函数来检查所有字母的这种情况：

["charge {}".format(sample_id) for sample_id in samples_to_check if 
 all(maximum[key] - val >= 0  for key, val in samples[sample_id].items())]

但它的可读性不如我推荐的普通 for 循环。

for sample_id in samples_to_check:
    if all(maximum[key] - val >= 0  for key, val in samples[sample_id].items()):
        print('sample_id:', sample_id)

Answer 2

与@Relandom 非常相似，但格式略有不同。我认为列表推导比常规的 for 循环更简洁，并且创建了更多的 Python 代码。

在这里，我不是循环遍历samples_to_check，而是遍历主要数据，然后检查sample_id 是否需要检查，这对我来说更有意义。

[
    f"CHARGE {sample_id}"
    for sample_id, sample in samples.items()
    if (
        sample_id in samples_to_check
        and all(maximum[key] >= val for key, val in sample.items())
    )
]