我想要一個函式,f給定一個 (n, 1) numpy 陣列回傳一個 (n, 1) numpy 陣列,其中的值是原始值加上它們的 0 軸索引。
例如:
f(np.array([[0],[0]])) = np.array([[0],[1]])
f(np.array([[4],[2],[3]])) = np.array([[4],[3],[5]])
是否有一些命令cmd(x)回傳函式正在操作的值的索引?的代碼f如下所示:
def f(x):
return x cmd(x)
uj5u.com熱心網友回復:
如果您想要無論 的形狀如何都可以作業的東西x,您想要類似的東西:
np.indices(x.shape)[0] 1
如果您使用的是大型陣列,則可以對其進行優化以構建較小的陣列。
轉載請註明出處,本文鏈接:https://www.uj5u.com/qukuanlian/483011.html
