更改列表中的一个权重并相应地调整所有其他权重，以便列表的总和再次为 1.0

Question

我有一个权重列表，它们的值范围都在 0.0 到 1.0 之间。 列表中值的总和应始终为 1.0。

现在我想写一个 function，我可以在其中将列表中的一个权重更改为某个值（正或负）。 lst 的剩余权重应该均匀调整，以便列表的总和在最后再次产生 1.0。

例子：

weights = [0.5, 0.2, 0.2, 0.1]

如果我将列表的第二个条目增加 0.3，则结果列表应如下所示：

weights = [0.4, 0.5, 0.1, 0.0]

我试过以下 function：

def change_weight(weights, index, value):
  result = []
  weight_to_change = weights[index] + value
  weights.pop(index)
  for i, weight in enumerate(weights):
    if i == index:
      result.append(weight_to_change)

    result.append(weight - value/len(weights))

  return result

这非常适合上面的示例：

weights = [0.5, 0.2, 0.2, 0.1]
print(change_weight(weights, 1, 0.3))
# like expected: [0.4, 0.5, 0.1, 0.0]

但是，如果我想将第二个权重更改为 0.5 左右。 列表的最后一个元素将得到一个负值：

weights = [0.5, 0.2, 0.2, 0.1]
print(change_weight(weights, 1, 0.5))

结果为[0.33, 0.7, 0.03, -0.07]

但是，我不希望列表中有任何负值。 这些值应该改为设置为 0.0，余数均匀地添加或减去其他值。

有谁知道我如何实现这个？

Answer 1

这是 @RemiCuin.net 的想法的实现：

def change_weight(weights, index, value):
    new_weight = weights[index] + value
    old_sum = sum(w for i,w in enumerate(weights) if i != index)
    new_weights = []
    for i,w in enumerate(weights):
        if i == index:
            new_weights.append(new_weight)
        else:
            new_weights.append(w*(1-new_weight)/old_sum)
    return new_weights

例如

print(change_weight([0.5, 0.2, 0.2, 0.1],1,.3))
print(change_weight([0.5, 0.2, 0.2, 0.1],1,.5))

Output：

[0.3125, 0.5, 0.12500000000000003, 0.06250000000000001]
[0.18750000000000006, 0.7, 0.07500000000000002, 0.03750000000000001]