我查看了與使用另一個 numpy 陣列索引 numpy 陣列相關的其他帖子,但仍然無法完成以下操作:
a = [[[1,2,3],[4,5,6]],[[7,8,9],[10,11,12]]],
b = [[[1,0],[0,1]],[[1,1],[0,1]]]
a[b] = [[[7,8,9],[4,5,6]],[[10,11,12],[4,5,6]]]
a是由 3D numpy 陣串列示的影像,維度為 2 * 2 * 3,最后一個維度為 RGB 值。b包含將與影像匹配的索引。例如對于像素索引 (0,0),它應該映射到原始影像的索引 (1,0),它應該給出像素值 [7,8,9]。我想知道是否有辦法實作這一目標。謝謝!
uj5u.com熱心網友回復:
這是一種方法:
In [54]: a = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])
In [55]: b = np.array([[[1, 0], [0, 1]], [[1, 1], [0, 1]]])
In [56]: a[b[:, :, 0], b[:, :, 1]]
Out[56]:
array([[[ 7, 8, 9],
[ 4, 5, 6]],
[[10, 11, 12],
[ 4, 5, 6]]])
轉載請註明出處,本文鏈接:https://www.uj5u.com/net/334210.html
下一篇:沿自己的路徑移動點
