Python中按匹配次数拆分查询结果并转map写入CSV的问题

看起来你遇到的核心问题是没有按行拆分多条记录，而是把所有内容一次性按|分割，导致换行符混入了字段中。咱们一步步来修复这个问题：

问题根源 #

当匹配数>1时，你的代码只截取了第一个换行后的所有内容，直接用split('|')处理，这会把换行符\n留在字段里（比如'Z\nid'），最终得到错误的数组结构。正确的思路应该是先把每条记录按行分开，再逐行处理字段。

解决方案步骤 #

1. 正确拆分所有记录行 #

首先把返回的完整字符串按换行符分割成单独的行，同时过滤掉空行（避免最后一行空行干扰）：

# 拆分所有行，过滤空行并去除每行首尾空白
all_lines = [line.strip() for line in data.value.split('\n') if line.strip()]
# 去掉表头行（如果你的结果始终带表头）
record_lines = all_lines[1:]

2. 逐行处理字段 #

对每一行记录单独按|分割，同时处理掉每行末尾可能存在的空字段（因为你的示例里每行最后有个|，会生成空字符串）：

processed_records = []
for line in record_lines:
    # 分割字段并去除每个字段的空白
    fields = [field.strip() for field in line.split('|')]
    # 过滤掉空字段（解决末尾|带来的空值）
    cleaned_fields = [f for f in fields if f]
    processed_records.append(cleaned_fields)

这样处理后，不管是1条还是N条记录，都会得到格式统一的二维数组，比如：

[['id1', 'K1', 'X1', 'D1', 'Z1'], ['id2', 'K2', 'X2', 'D2', 'Z2'], ['id3', 'K3', 'X3', 'D3', 'Z3']]

3. 规范写入CSV #

建议用Python内置的csv模块来写入，避免手动拼接字符串可能出现的格式问题（比如字段包含逗号的情况）：

import csv

# 写入CSV文件
with open('output.csv', 'w', newline='', encoding='utf-8') as csv_file:
    # 创建CSV写入器
    writer = csv.writer(csv_file)
    # 写入表头（如果需要的话）
    writer.writerow(['id', 'K', 'X', 'D', 'Z'])
    # 批量写入所有处理好的记录
    writer.writerows(processed_records)

完整示例代码 #

把上面的步骤整合起来，完整代码如下：

import csv

# 模拟返回的数据库结果
data = type('obj', (object,), {'value': """HEADER1|HEADER2|HEADER3|HEADER4|HEADER5
id1|K1|X1|D1|Z1|
id2|K2|X2|D2|Z2|
id3|K3|X3|D3|Z3|"""})()

# 处理数据
all_lines = [line.strip() for line in data.value.split('\n') if line.strip()]
record_lines = all_lines[1:]

processed_records = []
for line in record_lines:
    fields = [field.strip() for field in line.split('|')]
    cleaned_fields = [f for f in fields if f]
    processed_records.append(cleaned_fields)

# 写入CSV
with open('output.csv', 'w', newline='', encoding='utf-8') as csv_file:
    writer = csv.writer(csv_file)
    writer.writerow(['id', 'K', 'X', 'D', 'Z'])
    writer.writerows(processed_records)

这样不管匹配数是1还是多条，都能正确处理并写入CSV啦！

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