例如，我有：
arr = np.array[[10 30 20],
[30 20 10]]
indices = np.array[[2 1 0],
[2 1 0]]
我想要：
[[20 30 10],
[10 20 30]]
非常感謝！！
uj5u.com熱心網友回復：
使用np.take_along_axis：
import numpy as np
arr = np.array([[10, 30, 20],
[30, 20, 10]])
indices = np.array([[2, 1, 0],
[2, 1, 0]])
res = np.take_along_axis(arr, indices, axis=1)
print(res)
輸出
[[20 30 10]
[10 20 30]]
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/344674.html
上一篇：為什么我們需要在更改列后重新定義pandasDataFrame？
下一篇：Numpy：按列折疊陣列