用numpy求转移矩阵,可以这样做

要想求转移矩阵的话,最为常见的做法就是遍历所有的要素,需要比较长的时间,也需要比较长的代码,但是用numpy的话,三行代码就可以解决问题了。

事先知道分类类型的话,假设有a、b矩阵,求出各种类型在a、b矩阵中的索引,然后交叉,即可得到符合某类转移矩阵类别的索引,直接求数量就好了。

但是,要记得flatten()啊。

pos1=np.where(a==category1)
pos2=np.where(b==category2)
info=np.intersect1d(pos1,pos2).size

猜你喜欢

转载自blog.csdn.net/esa72ya/article/details/105198107