MATLAB与numpy矩阵中元素位运算的实现区别

最新推荐文章于 2022-12-06 21:15:23 发布

清风醉雨

最新推荐文章于 2022-12-06 21:15:23 发布

阅读量99

点赞数

分类专栏： PY与MATLAB 文章标签： matlab python

本文链接：https://blog.csdn.net/Firewall5788/article/details/120277545

版权

PY与MATLAB 专栏收录该内容

6 篇文章 1 订阅

订阅专栏

该博客讨论了如何在MATLAB代码转换为Python时处理二值矩阵的逻辑运算。问题集中在找出多个矩阵同时为1的元素下标。MATLAB中使用`find`函数，而Python则通过`numpy`库的`logical_and`和`nonzero`函数实现相同功能，注意结果的扁平化和索引获取。

摘要由CSDN通过智能技术生成

缘由：在将MATLAB代码转换成Python代码时所遇到的问题。

存在m*n二值矩阵（逻辑矩阵，元素由0和1组成mat1,mat2,mat3,mat4,求多个矩阵中同时大于0的位置下标(MATLAB中矩阵的下标以列为主序，即从第一列的第一行到最后一行依次编号，接着第二列，。。。)

MATLAB代码为：

index = find(mat1 & mat2 & mat3 & mat4);

Python代码为：

import  numpy as np
temp = np.logical_and(np.logical_and(np.logical_and(mat1, mat2), mat3),mat4).flatten(1) 
#flatten(1)表示结果以列为主序进行扁平化
index = np.nonzero(temp)[1] #因为已经扁平化成一行了，所以此处去不为0的元素的列坐标