如何用awk或Python实现多文件索引匹配与数值筛选输出？

当然可以用awk或者Python轻松实现这个需求！下面分别给出两种工具的解决方案，附详细解释和示例验证。

一、用awk实现 #

awk处理文本文件的对应行和列非常方便，不需要额外依赖，直接一行命令就能搞定：

BEGIN {OFS=" "}
{
    getline f2 < "File2"
    getline f3 < "File3"
    split(f2, arr2)
    split(f3, arr3)
    for(i=1; i<=NF; i++) {
        if ($i > 0 && $i < 100 && arr2[i] > 0 && arr2[i] < 10) {
            printf "%s ", arr3[i]
        } else {
            printf "0 "
        }
    }
    print ""
}

你可以把上面的代码保存为filter.awk，然后运行命令：

awk -f filter.awk File1

代码解释： #

• BEGIN {OFS=" "}：设置输出分隔符为空格，保证输出格式和输入一致
• getline f2 < "File2"：每读取一行File1的内容，同步读取File2的对应行
• split(f2, arr2)：把File2的行拆分成数组，方便按列索引取值
• 循环遍历每一列，判断两个条件：File1当前值在(0,100)区间，且File2对应值在(0,10)区间
• 满足条件就输出File3的对应值，否则输出0；最后换行保证行结构匹配

二、用Python实现 #

Python的灵活性更高，适合处理更复杂的扩展需求，这里提供两种实现方式：

1. 纯Python（无第三方依赖） #

如果不想安装额外库，用标准库就能完成：

with open("File1") as f1, open("File2") as f2, open("File3") as f3:
    # 同步读取三个文件的对应行
    for line1, line2, line3 in zip(f1, f2, f3):
        # 把每行字符串拆分成数值列表
        cols1 = list(map(float, line1.strip().split()))
        cols2 = list(map(float, line2.strip().split()))
        cols3 = list(map(float, line3.strip().split()))
        
        result = []
        # 逐列判断条件
        for c1, c2, c3 in zip(cols1, cols2, cols3):
            if 0 < c1 < 100 and 0 < c2 < 10:
                # 保留File3的值，整数转成字符串更贴合示例格式
                result.append(str(int(c3)) if c3.is_integer() else str(c3))
            else:
                result.append("0")
        
        # 输出结果行
        print(" ".join(result))

2. Pandas（高效处理大数据量） #

如果你的文件有数百行甚至更多，用pandas会更高效，代码也更简洁：

import pandas as pd

# 读取三个文件，指定分隔符为任意空白字符，无表头
df1 = pd.read_csv("File1", sep="\s+", header=None)
df2 = pd.read_csv("File2", sep="\s+", header=None)
df3 = pd.read_csv("File3", sep="\s+", header=None)

# 创建布尔掩码：标记满足条件的位置
mask = (df1 > 0) & (df1 < 100) & (df2 > 0) & (df2 < 10)

# 应用掩码：满足条件保留File3的值，否则设为0
result = df3.where(mask, 0)

# 输出结果，保持空格分隔的格式
print(result.to_csv(sep=" ", header=False, index=False))

示例验证 #

示例输入：

File1内容：

-10 -10 9 -20 88 106 -30 300 120 

File2内容：

-6 0 -7 -5 6 1 -2 18 32 

File3内容：

4 3 5 6 8 8 10 23 14 

预期输出：

0 0 0 0 8 0 0 0 0 

运行上面的awk或Python代码，都能得到这个预期结果。

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