如果我有 a 和 b:
a=[[1,2,3],
[4,5,6],
[7,8,9]]
b=8.1
我想在a中找到值b的索引,我可以这样做:
nonzero(abs(a-b)<0.5)
得到(2,1)作为索引,但是如果b是一维或者二维数组怎么办?说,
b=[8.1,3.1,9.1]
我想得到 (2,1),(0,2),(2,2)
一般来说,对于 b 的每个值,我希望 a 中只有一个匹配项。我可以避免 for 循环吗?
最佳答案
使用列表理解:
[nonzero(abs(x-a)<0.5) for x in b]
关于Python:查找索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/34687530/