我正在处理以下问题:我有一个列表，比方说a=[1, 2, 3]和一个数组b=[[2, 4, 6],[3, 2, 5],[4, 1, 3]]，列表中元素的数量等于数组中的行数，输出应该是列表c=[-1, 1, 2]，其中-1表示在数组b的第0行中没有找到数字a[0]，其他数字告诉索引元素a[i]在行b[i, :]中的什么位置，索引枚举从0开始.
我使用循环解决了它，但我正在寻找一种更优的方法来使用np.where或Torch替代方案，如果有任何帮助，我将非常感激.