在matlab中,如果A是数组,Index是索引数组,那么A(Index)是一种快速的根据索引Index对数组A直接进行重新排列的方法。
这种方法的好处就是得到索引后,对相关的矩阵,便可以直接进行重排。我在图像处理混合高斯背景建模中,由于对每个像素点都需要通过3-5个高斯模型进行计算比较,所以会有多个三维矩阵共用一个索引矩阵的情况。
但是,在matlab中,这种操作是如何进行的?如何在python中也能得到同样效果?
比如:(CSDN没有matlab的代码高亮格式,(╯﹏╰)b)
rand('seed',0);
array =randi(10,2,3,4); % 返回2*3*4的三维矩阵,每个元素都是1-10的随机整数
[sorted_array, sorted_index]=sort(array,3);
% 将array按照维度3升序排列,sorted_array是升序排列后的矩阵,
sorted_index是sorted_array中的元素在第三维度上对应于array中的位置
此时,因为设置了seed,所以尽管是随机数矩阵,但每次调试结果都是一样的。
三个数组应该如下: