要想求转移矩阵的话,最为常见的做法就是遍历所有的要素,需要比较长的时间,也需要比较长的代码,但是用numpy的话,三行代码就可以解决问题了。
事先知道分类类型的话,假设有a、b矩阵,求出各种类型在a、b矩阵中的索引,然后交叉,即可得到符合某类转移矩阵类别的索引,直接求数量就好了。
但是,要记得flatten()啊。
pos1=np.where(a==category1)
pos2=np.where(b==category2)
info=np.intersect1d(pos1,pos2).size