如何在awk循环前选择要遍历的目标数组？

问题核心 #

你需要根据变量值选择两个数组之一进行遍历，不想重复编写for循环内的大量处理代码，同时希望避免额外遍历开销。之前尝试的三元运算符、直接数组赋值、函数传数组的方法都因标准awk的语法限制报错。

标准awk的限制说明 #

标准awk不支持以下操作：

• 三元运算符返回数组（它是标量运算符，会将数组强制转为标量导致报错）
• 直接将一个数组赋值给另一个变量（同样会把数组当作标量处理）
• 直接将数组作为函数参数传递（会触发“尝试在标量上下文使用数组”的错误）

可行解决方案 #

利用标准awk的数组间接引用特性——通过字符串形式的数组名来访问数组元素，就能实现复用处理逻辑的目标。

方案1：封装处理逻辑到函数，传递数组名字符串 #

把遍历和处理逻辑写进函数，调用时根据变量值传递对应数组的名字符串：

function process_arr(arr_name,   i) {
    # 这里通过字符串形式的数组名间接访问元素
    for (i in arr_name) {
        # 替换成你的实际处理逻辑
        print "处理元素：索引" i "，值" arr_name[i]
    }
}

# 在主逻辑中根据条件调用函数
END {
    # 示例数组
    array1[1]="abc"
    array1[2]="def"
    array2[1]="ABC"
    array2[2]="DEF"
    
    independent_variable=0
    if (independent_variable) {
        process_arr("array1")
    } else {
        process_arr("array2")
    }
    
    independent_variable=1
    if (independent_variable) {
        process_arr("array1")
    } else {
        process_arr("array2")
    }
}

方案2：用变量存储数组名字符串，直接复用for循环 #

如果不想封装函数，也可以用变量存数组名，直接在同一个for循环中处理：

END {
    array1[1]="abc"
    array1[2]="def"
    array2[1]="ABC"
    array2[2]="DEF"
    
    var arr_name
    independent_variable=0
    # 根据条件选择数组名字符串
    arr_name = independent_variable ? "array1" : "array2"
    
    # 复用同一个for循环处理选中的数组
    for (i in arr_name) {
        print "处理元素：索引" i "，值" arr_name[i]
    }
}

验证修改后的测试代码 #

将你提供的测试代码改成使用数组间接引用的形式，就能正常运行：

function do_loop(arr_name,   i) {
    for (i in arr_name) {
        print "within function: " i " -> " arr_name[i]
    }
}

END {
    array1[1]="abc"
    array1[2]="def"
    array2[1]="ABC"
    array2[2]="DEF"
    for (i in array1) {
        print "outside function, array1: " array1[i]
    }
    for (j in array2) {
        print "outside function, array2: " array2[j]
    }
    
    independent_variable=0
    print "\nwith independent_variable == " independent_variable
    do_loop("array2")
    
    independent_variable=1
    print "\nwith independent_variable == " independent_variable
    do_loop("array1")
}

执行awk -f ternary_exp_for.awk <<< "test"后，会输出正常的遍历结果，不再报错。

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