我有两个 numpy 数组：a and b。我想选择所有索引a == 1 and b == 0.
也就是说，如果我有以下数组：
a = [0, 1, 3, 5, 1, 1, 2]
b = [1, 0, 2, 5, 3, 0, 6]
我想取回以下索引：
[1, 5]
我应该如何在 numpy 中执行此操作？我尝试使用以下内容（由显示 numpy、matlab 和 IDL 之间差异的快速参考指南建议）：
(a == 1 and b == 0).nonzero()
但这会产生关于真值不明确的错误。
有任何想法吗？
In []: from numpy import array, logical_and
In []: a= array([0, 1, 3, 5, 1, 1, 2])
In []: b= array([1, 0, 2, 5, 3, 0, 6])
In []: logical_and(a== 1, b== 0).nonzero()[0]
Out[]: array([1, 5])
显然这也可以工作：
In []: ((a== 1)& (b== 0)).nonzero()[0]
Out[]: array([1, 5])
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
