求C语言中替代Fortran WHERE语句的无循环数组操作方法

好问题！C语言本身并没有像Fortran WHERE 那样直接的内置条件批量操作语句，但我们有几种方式可以实现类似的功能——既能避免手动写繁琐的逐元素循环，又能让代码高效处理大型数值数组。下面是几个实用的方案：

GNU C向量扩展（GCC/Clang专属） #

如果你用的是GCC或Clang编译器，可以利用它们的向量扩展特性，直接定义向量类型进行批量条件判断和赋值，手动模拟WHERE的向量化逻辑：

#include <math.h>

// 定义256位双精度向量（包含4个double元素，适配AVX指令集）
typedef double v4df __attribute__ ((vector_size (32)));

void zero_small_elements(double *array, size_t n) {
    const v4df threshold = {1e-10, 1e-10, 1e-10, 1e-10};
    const v4df zero = {0.0, 0.0, 0.0, 0.0};
    size_t i;

    // 处理可以对齐到向量长度的批量元素
    for (i = 0; i <= n - 4; i += 4) {
        v4df vec = *(v4df *)&array[i];
        v4df abs_vec = __builtin_fabs(vec);
        // 条件选择：满足阈值条件的元素替换为0，否则保留原值
        vec = (abs_vec <= threshold) ? zero : vec;
        *(v4df *)&array[i] = vec;
    }

    // 处理剩余的零散元素
    for (; i < n; i++) {
        if (fabs(array[i]) <= 1e-10) {
            array[i] = 0.0;
        }
    }
}

这里的循环会被编译器优化成SIMD指令，相当于后台做了批量的WHERE操作，不需要你逐元素处理。

OpenMP SIMD 跨编译器方案 #

如果需要跨编译器兼容，OpenMP的SIMD指令是个不错的选择——给普通循环加个编译指令，编译器会自动帮你实现向量化，写法非常直观：

#include <math.h>
#include <omp.h>

void zero_small_elements(double *array, size_t n) {
    // 告诉编译器自动向量化这个循环
    #pragma omp simd
    for (size_t i = 0; i < n; i++) {
        if (fabs(array[i]) <= 1e-10) {
            array[i] = 0.0;
        }
    }
}

这个写法看起来是普通循环，但开启OpenMP并启用优化后，编译器会把它转换成类似Fortran WHERE的批量操作，性能拉满的同时代码还很简洁。

第三方数值库（零循环极简写法） #

如果你的项目本来就用数值计算库，比如ArrayFire、Eigen这类，它们提供了声明式的掩码赋值，完全不需要写循环，和Fortran WHERE的用法几乎一致：
以ArrayFire为例（支持CPU/GPU加速）：

#include <arrayfire.h>

void zero_small_elements_af(double *array, size_t n) {
    af::array arr(n, array);
    // 直接用掩码选择元素并赋值，和Fortran WHERE逻辑完全对应
    arr(af::abs(arr) <= 1e-10) = 0.0;
    arr.host(array); // 把处理后的数组拷回CPU内存
}

这种方式最接近你熟悉的Fortran语法，而且库会自动处理底层的并行和向量化，适合超大型数组的处理。

另外补充一句：现代编译器的优化能力很强——哪怕你写了普通的for循环，只要开启-O3这类优化选项，编译器通常会自动完成向量化优化，性能和上面的方案差距不大。但如果是追求和WHERE一致的声明式写法，上面的后两种方案会更贴合你的需求。

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