这是我的数组：
a = [[0.,0.,0.1,0.2], [0.,0.3,0.4,0.3], [0.,0.,0.1,0.]]
我想做一个where子句，它将返回'a'中元素的索引，其中该元素的值之和等于1。 where(sum(a) == 1)
where(sum(a) == 1)
有人可以指导我吗？
谢谢。
使用枚举，列表理解：
>>> a = [[0.,0.,0.1,0.2], [0.,0.3,0.4,0.3], [0.,0.,0.1,0.]] >>> [i for i, xs in enumerate(a) if sum(xs) == 1] [1]
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
点击生成二维码
我来说两句