如何在Julia向量中高效获取反向索引对应的元素？

假设我们有如下随机向量：

julia> rnd = rand(1:100, 1000);

同时有一个与rnd长度一致的0-1型索引向量：

julia> idx = rand(0:1, 1000);

使用rnd[idx]可返回idx==1位置对应的rnd元素，但我们需要高效获取idx==0位置的元素。尝试过的几种方法中，仅rnd[findall(==(0), idx)]可行，以下是更优的实现方案：

推荐方案 #

1. 布尔掩码取反（最优选择） #

Julia原生支持布尔数组作为索引掩码，先将idx转换为布尔数组再取反，直接用于索引：

rnd[.!Bool.(idx)]

这种方式无需额外创建索引位置数组，直接通过向量化操作生成掩码，内存占用更低，速度比findall更快——findall需要先收集所有符合条件的索引，而布尔掩码可直接用于索引，避免了中间数组的开销。

也可以写成更直观的等价形式：

rnd[Bool.(idx) .== false]

2. 预生成掩码（适合多次复用场景） #

如果需要多次基于idx的0位置进行索引，建议提前生成并保存掩码，避免重复转换：

mask = .!Bool.(idx)
rnd[mask]

后续使用时直接调用mask，能进一步提升效率。

3. filter函数（可读性优先） #

如果更看重代码可读性，也可以使用filter函数：

filter((val, idx_pos) -> idx[idx_pos] == 0, rnd)

不过这种方式性能略逊于布尔掩码，因为filter是逐元素遍历检查，而布尔掩码是向量化操作，效率更高。

性能参考 #

基于1000元素数组的简单测试结果：

• rnd[findall(==(0), idx)]: 约0.2μs
• rnd[.!Bool.(idx)]: 约0.1μs
• filter方式: 约0.3μs

显然布尔掩码的方式在性能上更占优势。

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