如何实现下面的功能？

Question

像下图这样从二维数组中随机抽取一个值，如果你知道检索到的值所在的列，就可以知道下一行。想知道代码如何实现。

twoarr = np.array(
  [
    ['A' , 'B' , 'C'],
    ['D' , 'E' , 'F'] ,
    ['G' , 'H' , 'I' ]
  ]
)

例如，如果值为 A，由于 A 是第一列中的值，因此将第一行中的“A，B，C”中的一个与下一个对应的 1 连接起来。
如果下一个值是B，那么B就是第二列的值，那么接下来连接对应于2的第二行的“D,E,F”之一。

当 n=2 时，例如 AA、BE、GB 等。
当 n=3 时，例如 AAA、DBF、HDA 等。

Answer 1

感觉问题描述的不清晰，看不懂需求