Python使用random.choices采样命名元组时如何获取原列表对应索引

你当前的写法是直接对self.memory的元素执行采样，所以random.choices返回的是采样到的Transition对象列表，而非这些对象在原列表中的整数下标，自然无法直接作为列表索引使用。

最优解决方案 #

直接对memory的下标范围执行带权重采样，修改代码如下：

# 对下标序列执行采样，权重和原逻辑完全匹配
sample_indices = random.choices(range(len(self.memory)), k=batch_size, weights=sample_probs)
# 按需获取采样到的Transition对象
sample_data = [self.memory[idx] for idx in sample_indices]

注意事项 #

• 不要用list.index()手动将采样到的Transition对象转回下标：如果memory中存在重复的Transition实例，index()只会返回第一个匹配项的下标，和实际采样位置不符，会导致逻辑错误
• 该方案的采样分布和你原有写法完全一致，不会改变采样概率逻辑

内容的提问来源于stack exchange，提问作者Peter K