我有两个numpy列表：
x = ['A', 'A', 'C', 'A', 'V', 'A', 'B', 'A', 'A', 'A']
y = ['1', '2', '1', '1', '3', '2', '1', '1', '1', '1']
同时x相等'A'和y相等时如何查找索引'2'？
我希望得到索引[1, 5]。
我尝试使用：np.where(x == 'A' and y == '2')但这并没有帮助我。
您需要将列表转换为numpy数组，才能使用向量化操作，例如==和&：
import numpy as np
np.where((np.array(x) == "A") & (np.array(y) == "2"))
# (array([1, 5]),)
较短的版本（如果您确定x和y是numpy数组）：
>>> np.where(np.logical_and(x == 'A', y == '2'))
(array([1, 5]),)
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句