作者flarehunter (Range)
看板Python
标题Re: [问题] 在任意维度的 List 中找某元素的位置？
时间Wed Nov 13 20:33:34 2019
※ 引述《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://webptt.com/cn.aspx?n=bbs/Python/M.1573648416.A.B01.html
1F：推 abram: 太强了 谢谢 学到一课！ 11/13 21:04
2F：推 abram: 这样就能处理任何维度矩形和非矩形的list了 赞 11/13 21:11
3F：推 abram: 没想到可以在Find的定义里面使用Find 太黯然太销魂了 11/13 21:13
4F：推 Ryspon: 推 11/13 22:29
5F：推 disney82231: 有点不太懂为什麽可以在Find里面使用Find，有大大可 11/13 22:56
6F：→ disney82231: 以解释吗 11/13 22:56
7F：推 abram: 回楼上 函式用自我呼叫的方式完成递回 11/13 23:11
9F：推 bdx915: To iterate is human ,To recurse is divine 11/14 09:09
10F：推 bookstar07: 这篇回文让我学到好多小技巧... 11/15 23:38
11F：→ bookstar07: 原本以为小懂python 结果发现新大陆XD 11/15 23:38
12F：推 a110482: 可以查关键字"recursive" 12/08 16:39