如果满足条件，则列出更新值的理解

Question

我有以下列表和变量I：

probs = [1/9, 1/9, 1/9, 1/9, 1/9, 1/9, 1/9, 1/9, 1/9]
i = 3

我想将变量I中给出的索引列表中的值更改为3/9，其余值更改为8/9。 因此，上述条件下的预期结果为：

probs = [8/9, 8/9, 8/9, 3/9, 8/9, 8/9, 8/9, 8/9, 8/9]

我更喜欢用列表理解来提高效率，有什么想法吗

Answer 1

您可以使用列表中的表达式来执行此操作

probs = [3 / 9 if i == index else 8 / 9 for index, _ in enumerate(probs)]