不好意思,因為用習慣了 Mathematica 的 Position 指令,
所以小弟正在用 Python 寫一個執行相同功能的指令。
現在卡住的點是,如果知道輸入的 List 是二維的,
當然就是用兩層的 for loop:
def position(arr,cri):
index=[];
for i in range(len(x)):
for j in range(len(x[i])):
if x[i][j] == cri:
index.append([i, j])
return index
相對地,若知道輸入 List 是三維的當然就用三層的 loop。
可是當程式處理的維度由輸入 List 內生決定時,例如在
Mathematica 可以執行:
Position[{2, {1, 2}, {{1, 1, 2}}}, 2]
得到結果為:
{{1}, {2, 2}, {3, 1, 3}}
不知道要怎麼寫才能在 Python 下實作類似的功能呢?
謝謝!