返回變異列表的表達式

Question

我正在尋找一個表達式，改變一個元素並返回修改后的列表

下面有點啰嗦

# key=0; value=3; rng=[1,2]
[(v if i != key else value) for i, v in enumerate(rng)]

---

編輯：

我正在尋找一種在單個表達式中內聯以下 function 的方法

def replace(rng: List, key: int, value):
    a = list(rng) 
    a[key] = value
    return a

---

編輯 2 ：真正激發這個問題的代碼

class TextDecoder(nn.Module):
    def forward(self, x: Tensor, kv_cache: Tensor):
        kv_cache_write = torch.zeros((_:=list(kv_cache.shape))).__setitem__(-2, x.shape[-1]) or _)
        ...

Answer 1

也許比列表串聯更好：

[*rng[:key], value, *rng[key+1:]]

其他：

[a for a, a[key] in [(rng[:], value)]][0]

或者，如果您只是將結果分配給一個變量（如前所述），您可以這樣做：

a, a[key] = rng[:], value

Answer 2

嘗試列表串聯：

key = 0
value = 3
rng = [1, 2]

out = rng[:key] + [value] + rng[key+1:]
print(out)

rng[:key]是列表的副本，直到鍵（不包括）， [value]是一個新列表，其中唯一的元素是value ，而rng[key+1]是從鍵開始的列表的副本（獨家的）。 將這些連接在一起，您將獲得一個副本，其中替換了密鑰。

Answer 3

# key=0; value=3; rng=[1,2]
print(rng.__setitem__(key, value) or rng)

如果你不想修改原始列表rng ，你可以這樣做

print((_:=list(rng)).__setitem__(key, value) or _)

不幸的是，這會泄漏變量_ 。