[问题] 在任意维度的 List 中找某元素的位置?

楼主: abram (科科)   2019-11-13 19:18:29
不好意思,因为用习惯了 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 下实作类似的功能呢?
谢谢!
作者: s860134 (s860134)   2019-11-13 19:42:00
itertools 弄成一维
楼主: abram (科科)   2019-11-13 19:52:00
弄成一维只能知道是否含此元素 无法知道位置
作者: s860134 (s860134)   2019-11-13 20:04:00
要找index 那用递回呼叫就可以了 看要 bfs 还dfs
作者: TitanEric (泰坦)   2019-11-13 20:09:00
可以用numpy搭配where
楼主: abram (科科)   2019-11-13 20:51:00
谢谢 argwhere对于任意维度的长方矩阵确实有用虽然对于非矩形的不对称list不适用 但也够用了
作者: nevak (^o^)   2019-11-13 21:06:00
可以用type()来判断是不是listex: https://gitlab.com/snippets/1912813 也可以用recursive更简洁一点

Links booklink

Contact Us: admin [ a t ] ucptt.com