CSV文件IP匹配脚本问题：A列IP需全量匹配B列而非同行对比

我来帮你搞定这个脚本的问题，你现在的核心需求是让A列的任意IP只要在B列所有行里出现过，就把该行导出到新CSV，同时还要排除B列空白行的干扰对吧？这就给你修正后的方案：

首先，我们需要先把B列所有非空白的IP地址收集到一个集合中（集合的查找效率远高于列表，适合快速做存在性检查），然后再逐行检查A列的IP是否在这个集合里，符合条件就写入新文件。

以下是完整的修正代码（假设你的CSV有表头，A列表头为"A"，B列表头为"B"）：

import csv

# 替换成你的原始CSV文件名和输出文件名
input_file = 'your_input.csv'
output_file = 'matched_results.csv'

# 第一步：读取原始文件，收集B列所有非空白的IP
b_ips = set()
with open(input_file, 'r', newline='', encoding='utf-8') as infile:
    reader = csv.DictReader(infile)
    for row in reader:
        # 去除IP前后的空白字符，避免空格导致的匹配失败
        b_ip = row['B'].strip()
        # 只收集非空白的IP，直接跳过B列空白的行
        if b_ip:
            b_ips.add(b_ip)

# 第二步：再次读取原始文件，筛选符合条件的行写入新文件
with open(input_file, 'r', newline='', encoding='utf-8') as infile, \
     open(output_file, 'w', newline='', encoding='utf-8') as outfile:
    reader = csv.DictReader(infile)
    # 保留原始表头写入新文件
    writer = csv.DictWriter(outfile, fieldnames=reader.fieldnames)
    writer.writeheader()
    
    for row in reader:
        a_ip = row['A'].strip()
        # 检查A列IP是否存在于B列的有效IP集合中
        if a_ip in b_ips:
            writer.writerow(row)

关键改动说明： #

• 全局匹配B列IP：先用一次遍历把B列所有有效的非空白IP存入集合b_ips，这样后续每一行的A列IP只需要做一次集合查找，效率比逐行对比高得多，也实现了A列IP匹配B列所有行的需求。
• 排除B列空白行：通过if b_ip:的判断，直接跳过B列空白的行，完全避免空白值干扰匹配逻辑。
• 更安全的文件操作：使用with语句自动管理文件的打开和关闭，不会出现资源泄漏；指定newline=''保证CSV文件的换行符在不同系统下都正确；用DictReader/DictWriter通过表头访问列，逻辑更清晰。

无表头版本（如果你的CSV没有表头，只有纯数据列） #

如果你的原始CSV没有表头，直接是A列（第一列）和B列（第二列）的IP数据，可以用这个版本：

import csv

input_file = 'your_input.csv'
output_file = 'matched_results.csv'

b_ips = set()
# 收集B列非空白IP
with open(input_file, 'r', newline='', encoding='utf-8') as infile:
    reader = csv.reader(infile)
    for row in reader:
        # 确保当前行至少有两列数据，避免索引越界
        if len(row) >= 2:
            b_ip = row[1].strip()
            if b_ip:
                b_ips.add(b_ip)

# 筛选符合条件的行
with open(input_file, 'r', newline='', encoding='utf-8') as infile, \
     open(output_file, 'w', newline='', encoding='utf-8') as outfile:
    reader = csv.reader(infile)
    writer = csv.writer(outfile)
    
    for row in reader:
        # 确保当前行有A列数据
        if len(row) >= 1:
            a_ip = row[0].strip()
            if a_ip in b_ips:
                writer.writerow(row)

这样就完美解决了你提到的所有问题：A列IP可以匹配B列任意一行的IP，同时完全忽略B列的空白行。

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