python - 减去 boolean 列表

Question

我有两个 boolean 值列表 buy_machine 和 broken_machine。我想创建第三个列表 working_machines，它是购买的机器数量的总和并减去坏机器的数量。

我尝试了以下代码，结果列表中充满了 'nothing' 我希望至少看到 'plus 1' 或 'minus 1 '，这只是为了尝试一些事情，我不知道如何实际制作 working_machines 列表来对 boolean 值求和或减去

years = list(range(1, 21))
buy_machine = []
broken_machine = []
working_machines = []
for year in years:
    if year <= 5:
        buy_machine.append(True)
    else:
        buy_machine.append(False)
    if 10 % year == 10 and year <= 15:
        broken_machine.append(True)
    else:
        broken_machine.append(False)
    if buy_machine == True and broken_machine == False:
        working_machines.append('plus 1')
    elif buy_machine == False and broken_machine == True:
        working_machines.append('minus 1')
    else:
        working_machines.append('nothing')

buy_machine
[True,
 True,
 True,
 True,
 True,
 False,
 False,
 False,
 False,
 False,
 False,
 False,
 False,
 False,
 False,
 False,
 False,
 False,
 False,
 False]

broken_machine
[False,
 False,
 False,
 False,
 False,
 False,
 False,
 False,
 False,
 False,
 True,
 True,
 True,
 True,
 True,
 False,
 False,
 False,
 False,
 False]

working_machines
['nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing',
 'nothing']

想要的输出

working_machines
[1, 2, 3, 4, 5, 5, 5, 5, 5, 5, 4, 3, 2, 1, 0, 0, 0, 0, 0, 0]

Answer 1

working_machines 只是以下列表中值的累加和:

>>> x = [a - b for a,b in zip(buy_machine, broken_machine)]
>>> x
[1, 1, 1, 1, 1, 0, 0, 0, 0, 0, -1, -1, -1, -1, -1, 0, 0, 0, 0, 0]

(这是有效的，因为 True 和 False 分别等于 1 和 0， bool 是 int 的子类。)

您可以使用 itertools.accumulate 获取它。

>>> list(accumulate(x))
[1, 2, 3, 4, 5, 5, 5, 5, 5, 5, 4, 3, 2, 1, 0, 0, 0, 0, 0, 0]

也就是说，对于每个索引 i，working_sums[i] = sum(x[:i])。