找到两个列表的所有组合

Question

I have a question regarding the python. I have a grid like the attached figure. I have the coordinates of all states. The coordinates of the states are shown in the A and B lists:

A = [1,2,3]
B = [4,5,6]

I want to find the coordinate of the state that I want. For example, I want to go to state (0,1) and I want its coordinate as y =(2,4).

Answer 1

My solution to this would be to create the following function:

def convert(state, A, B): # state should be of the shape (int, int)
   return (A[state[1]], B[state[0]])

There is no error checking here so make sure the indexes are within bounds.

You can now use the function as so:

A=[1,2,3]
B=[4,5,6]
sx = (0,1)
y = convert(sx, A, B)

I hope this helps

Answer 2

y=(A[1],B[0])

try list indexing
def getv(x,y):
  a = A.index(x)
  b = B.index(y)
  return (a,b)

getv(0,1)