访问列表的多个项目

问题描述

我目前正在尝试实现一个重播缓冲区，在其中我将 20 个数字存储在一个列表中，然后想要随机采样这些数字中的 5 个。

我用 numpy 数组尝试了以下操作：

ac = np.zeros(20,dtype=np.int32)

for i in range(20):
    ac[i] = i+1

batch = np.random.choice(20,5,replace=False)
sample = ac[batch]

print(sample)

这按它应该的方式工作，但我想用列表而不是 numpy 数组来做同样的事情。

但是当我尝试使用列表获取 sample = ac[batch] 时，我收到此错误消息：

TypeError: 只有整数标量数组可以转换为标量索引

如何像使用 numpy 一样访问列表的多个元素？

解决方法

对于列表，这很容易。只需使用随机模块中的示例函数：

import random

ac = [i+1 for i in range(20)]
sample = random.sample(ac,5)

另外附注：当你想创建一个包含一系列数字的 numpy 数组时，你不必创建一个带有零的数组，然后将它填充到 for 循环中，这不太方便，而且很显着比使用 numpy 函数 arange 慢。

ac = np.arange(1,21,1)

如果你真的想创建一个包含你想要访问的索引的批处理列表，那么你将不得不使用列表理解来访问这些，因为你不能像一个 numpy 数组那样只索引一个具有多个索引的列表。

batch = [random.randint(0,20) for _ in range(5)]
sample = [ac[i] for i in batch]