如何解决访问列表的多个项目
我目前正在尝试实现一个重播缓冲区,在其中我将 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]
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。