Re: [問題] 在任意維度的 List 中找某元素的位置？

看板Python作者flarehunter (Range)時間4年前 (2019/11/13 20:33)推噓9(9推 0噓 3→)

留言12則, 6人參與討論串2/2 (看更多)

※ 引述《abram (科科)》之銘言： : 不好意思，因為用習慣了 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 下實作類似的功能呢？ : 謝謝！如果是List的話就遞迴下去找，找到最後如果符合的話就回傳index def Find(arr, value, current_idx): if not isinstance(arr, list): if arr == value: return [current_idx] else: return [] ret = [] for idx, item in enumerate(arr): ret += Find(item, value, current_idx + [idx]) return ret arr = [2, [1, 2], [[1, 1, 2]]] print Find(arr, 2, []) # [[0], [1, 1], [2, 0, 2]] -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 126.74.152.122 (日本) ※ 文章網址: https://www.ptt.cc/bbs/Python/M.1573648416.A.B01.html

推

abram

11/13 21:04, 4年前 , 1^F

11/13 21:04, 1^F

推

abram

11/13 21:11, 4年前 , 2^F

11/13 21:11, 2^F

推

abram

11/13 21:13, 4年前 , 3^F

11/13 21:13, 3^F