python - 找到两个列表的所有组合

Question

我有一个关于 python 的问题。我有一个像附图一样的网格。我有所有州的坐标。状态的坐标显示在 A 和 B 列表中：

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

我想找到我想要的状态的坐标。例如，我想进入状态 (0,1)，我希望它的坐标为 y =(2,4)。

标签： pythonlist

Answer 1

我对此的解决方案是创建以下函数：

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)

我希望这有帮助