Python文件如何访问变量

Question

我使用以下tensorlayer命令创建了一个npy数据集。

tl.files.save_any_to_npy(
save_dict={
    'images': aggregated_images, 
    'actions': aggregated_actions,
    'rewards': aggregated_rewards}, 
    name='./data/episode0.npy')

我能够加载文件(奖励/操作是标量数组；图像是矩阵的数组)。

import numpy as np
data = np.load('./data/episode0.npy')

我认为这类似于字典(print(data)工作)。所以，我试过

actions = data['actions'] 

但是这给了我以下错误

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices
>>> actions = data['rewards']
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices

如何解决此错误？我想我可以使用三个变量来解决这个问题，但是我宁愿只跟踪一个文件。 解决方案(信用归Goyo): 导入张力层为tl data = tl.files.load_npy_to_any(path='./data'，name=‘asede0.npy’)操作=数据‘action’

Answer 1

试试这个：

data = np.load('./data/episode0.npy').item()
data["actions"]