我正在尝试使用numpy lib将一些matlab代码转换为python代码。 代码搜索数组值并保存找到的索引
原始的matlabcode
index1 = find(array1 == 2 & array2 > array3);
我的python“翻译”
index1 = np.where((array1 == 2) & (array2 > array3))
这是正确的方法吗？ 我无法测试输出，因为我没有matlab，我希望有人可以帮助我。谢谢！
答案 0 :(得分：1)
试
np.nonzero( np.logical_and( array1 == 2, array2 > array3 ) )
您可以详细了解logical_and和nonzero。您也可以考虑将nonzero()替换为flatnonzero