我有一个二维数组,如:
a = [[0, 0, 0],
[1, 1, 1],
[2, 2, 2],
[3, 3, 3]]
我的索引数组如下:
b = [[0,0,1],
[2,1,3]]
我的目标数组应该是:
c = [[[0,0,0],
[0,0,0],
[1,1,1]],
[[2,2,2],
[1,1,1],
[3,3,3]]]
如何使用NumPy包?是否有任何内置函数可以做到这一点? 谢谢你的帮助。