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