如何修复 python 中的这个索引错误？

Question

我的代码如下：

environment_rows = 3  #1
look_up_tables = [np.zeros((environment_rows, 20)) for _ in range(35)]



def get_starting_look_up_table(): #2
  current_look_up_table = np.random.randint(35)
  return current_look_up_table

def get_next_action(number_of_look_up_table,current_row_index,epsilon):
  if np.random.random() < epsilon:
    result=np.argmax(look_up_tables[number_of_look_up_table][current_row_index])
  else: #choose a random action
    result=np.random.randint(20)
  return result



number_of_look_up_table = get_starting_look_up_table() #3
current_row_index = 0
epsilon = 0.9
action_index = get_next_action(current_row_index,number_of_look_up_table,epsilon)

在第一部分，我生成了与look_up_tables 相关的矩阵，每个arrays 有3 行20 列。 我们总共有 35 个这样的 arrays。

在第二部分中，使用 'get_starting_look_up_table' function 随机选择 35 个中的一个。 此外，通过使用“获取下一个操作”function，应根据最大值或随机选择先前 function 选择的数组的 20 列之一。

最后，在第三部分调用了函数，但出现以下错误。 当我单独运行有错误的行时，我没有任何问题。 感谢您在这方面指导我。

IndexError                                Traceback (most recent call last)
~\AppData\Local\Temp/ipykernel_28528/1385508929.py in <module>
   2 current_row_index = 0
   3 
 ----> 4 action_index = get_next_action(current_row_index,number_of_look_up_table,epsilon)
   5 
   6 reward = determine_rewards(number_of_look_up_table,current_row_index,action_index)

  ~\AppData\Local\Temp/ipykernel_28528/255015173.py in 
   get_next_action(number_of_look_up_table, current_row_index, epsilon)
  9 def get_next_action(number_of_look_up_table,current_row_index,epsilon):
 10     if np.random.random() < epsilon:
---> 11      return np.argmax(look_up_tables[number_of_look_up_table][current_row_index])
    12     else: #choose a random action
    13      return np.random.randint(20)

  IndexError: index 27 is out of bounds for axis 0 with size 3

Answer 1

您的 function 调用get_next_action(current_row_index, number_of_look_up_table, epsilon)的参数顺序错误。 您切换current_row_index和number_of_look_up_table 。

我想你想要这个：

action_index = get_next_action(number_of_look_up_table, current_row_index, epsilon)