Мое решение для этого было бы создать следующую функцию:
def convert(state, A, B): # state should be of the shape (int, int)
return (A[state[1]], B[state[0]])
Здесь нет проверки ошибок, поэтому убедитесь, что индексы находятся в пределах границ.
Теперь вы можете использовать Функция так:
A=[1,2,3]
B=[4,5,6]
sx = (0,1)
y = convert(sx, A, B)
Надеюсь, это поможет