如何让Python二维掩码数组的布尔索引结果顺序与MATLAB一致

核心原因 #

• MATLAB数组默认采用*列优先（Fortran顺序）*存储，布尔索引时按列遍历匹配元素
• NumPy数组默认采用*行优先（C顺序）*存储，布尔索引时按行遍历匹配元素
  两者存储遍历顺序的差异直接导致了筛选结果的元素顺序不同。

小提示：你提供的示例Python代码中np.array定义末尾多写了多余的右括号，实际运行时需要删除避免语法错误。

实现方案 #

两种常用方案都可以得到和MATLAB完全一致的输出结果：

方法1：转置后索引 #

对原数组和布尔掩码同时做转置后再执行布尔索引，转置后的行优先遍历等效于原数组的列优先遍历：

import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
b = np.array([[False, False, True], [False, False, False], [True, False, False]])
res = a.T[b.T]
print(res) # 输出 [7 3]
# 如果需要和MATLAB一致的列向量格式，可补充reshape操作
res_col = res.reshape(-1, 1)

方法2：按列优先顺序展平后索引 #

通过order='F'参数指定按列优先（Fortran顺序）展平数组和掩码，再执行索引：

res = a.ravel(order='F')[b.ravel(order='F')]
print(res) # 输出 [7 3]

内容的提问来源于stack exchange，提问作者Feulo