如何修改硬编码字段的awk脚本，支持命令行传入目标字段参数？

现有一段awk脚本可从CSV文件中提取第6、7、14字段，脚本如下：

awk -F, '{for (i=1; i<=NF; i++) if (i in [6, 7, 14]) printf "%s,", $i; print ""}' $input_file

该脚本运行正常，但目标字段为硬编码形式。希望通过命令行参数传入目标字段以提升通用性，尝试修改后始终出现语法错误：

awk -F, '{for (i=1; i<=NF; i++) if (i in ['$2']) printf "%s,", $i; print ""}' $input_file

1. awk的in操作符要求匹配关联数组，你之前硬编码的[6,7,14]其实是不符合awk语法的，只是碰巧没触发报错。
2. 直接在awk代码里嵌入['$2']会导致shell解析混乱：如果传入多字段参数（比如6,7,14），shell会拆分参数，同时awk无法识别这种语法结构。

方案1：处理单个字段参数 #

如果只需要提取单个字段（比如运行脚本时传入./script.sh input.csv 6），用-v选项把字段号传给awk即可：

# 脚本示例（假设脚本名为extract_fields.sh）
input_file=$1
target_field=$2
awk -F, -v field="$target_field" '{for(i=1;i<=NF;i++) if(i==field) printf "%s,", $i; print ""}' "$input_file"

方案2：处理多个字段参数 #

如果需要传入多个字段，推荐把字段列表传入awk后转成关联数组，以下两种方式任选：

方式A：逗号分隔的字段列表作为参数 #

input_file=$1
fields_list=$2
awk -F, -v fields="$fields_list" '
BEGIN {
    # 把逗号分隔的字段列表拆成关联数组
    n = split(fields, arr, ",")
    for(i=1;i<=n;i++) target_fields[arr[i]] = 1
}
{
    for(i=1;i<=NF;i++) {
        if(i in target_fields) printf "%s,", $i
    }
    print ""
}' "$input_file"

运行示例：./script.sh data.csv 6,7,14

方式B：空格分隔的多个字段参数 #

如果习惯用空格传多个字段（比如./script.sh data.csv 6 7 14），可以利用awk的ARGV数组处理：

input_file=$1
shift # 移除第一个参数（文件名），剩下的都是字段号
awk -F, '
BEGIN {
    for(i=1;i<ARGC;i++) target_fields[ARGV[i]] = 1
    ARGC = 1 # 清空多余参数，避免awk把字段当成文件名
}
{
    for(i=1;i<=NF;i++) {
        if(i in target_fields) printf "%s,", $i
    }
    print ""
}' "$input_file" "$@"

额外优化：去掉末尾多余逗号 #

原脚本会在每行末尾留下多余逗号，可优化输出逻辑：

awk -F, -v fields="$fields_list" '
BEGIN {
    split(fields, arr, ",")
    for(i=1;i<=length(arr);i++) target_fields[arr[i]] = 1
}
{
    sep = ""
    for(i=1;i<=NF;i++) {
        if(i in target_fields) {
            printf "%s%s", sep, $i
            sep = ","
        }
    }
    print ""
}' "$input_file"

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