Python中匹配不同大小二维数组并生成新数组的问题

你的核心问题是错误地按数组的索引位置去匹配值，而不是根据文件A第一列的数值去查找文件B中对应的行。原代码里的x[i] == x1[i]是在比较两个数组相同位置的元素，但实际上你需要的是找出文件B中第一列数值存在于文件A第一列中的行，这就导致代码逻辑完全不符合需求。

完整解决方案 #

我们可以分三步来实现：读取文件数据、提取匹配基准、筛选目标行，下面是具体的Python代码实现：

1. 读取并解析文件数据

首先我们需要把两个文件的内容转换成可用的数据结构，同时跳过注释行：

# 从文件中提取第一列的所有数值（作为匹配基准）
def get_match_values(file_path):
    match_set = set()
    with open(file_path, 'r') as f:
        for line in f:
            stripped_line = line.strip()
            # 跳过空行和注释行
            if not stripped_line or stripped_line.startswith('#'):
                continue
            # 提取第一列的数值
            first_col = int(stripped_line.split()[0])
            match_set.add(first_col)
    return match_set

# 读取文件的所有行数据（保留完整的行内容）
def get_file_rows(file_path):
    rows = []
    with open(file_path, 'r') as f:
        for line in f:
            stripped_line = line.strip()
            if not stripped_line or stripped_line.startswith('#'):
                continue
            # 将每行转换成整数列表
            row = list(map(int, stripped_line.split()))
            rows.append(row)
    return rows

# 读取文件A的匹配基准和文件B的原始数据
a_match_values = get_match_values('文件A.txt')
b_raw_data = get_file_rows('文件B.txt')

2. 筛选符合条件的行

现在我们遍历文件B的所有行，只保留第一列数值在文件A匹配集合中的行：

# 筛选匹配的行
matched_rows = []
for row in b_raw_data:
    if row[0] in a_match_values:
        matched_rows.append(row)

3. 输出结果

最后按照要求的格式输出结果：

print("#newx newy")
for row in matched_rows:
    print(f"{row[0]} {row[1]}")

原代码的问题分析 #

你之前的代码存在几个关键问题：

• 没有明确x和x1的来源（应该是两个文件第一列的数值列表）
• 逻辑错误：x[i] == x1[i]是比较相同索引位置的元素，而不是检查x1中的值是否存在于x的集合里，这完全不符合你的匹配需求
• 只提取了x1[i]，没有保留对应的y1值，导致最终结果丢失了第二列的数据

示例运行结果 #

用你提供的测试数据，运行后会输出：

#newx newy
1 1
5 1
7 1

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