Python Pandas：将DataFrame数据按自定义块格式写入文件

// Lines of comments
identifier1       label2 = i  label3        label4                                  
label5
A1 = -5563.88 B2 = -4998 C3 = -203.8888 D4 = 5926.8 
E5 = 24.99876 F6 = 100.6666 G7 = 30.008 H8 = 10.9999
J9 = 1000000 K10 = 1.0002 L11 = 0.1
M12

identifier2       label2 = i  label3        label4                                  
label5
A1 = -788 B2 = -6554 C3 = -100.23 D4 = 7526.8 
...

import pandas as pd
import numpy as np
import string

# --- 示例数据生成 (用于演示，实际应用中替换为您的真实DataFrame) ---
# rng 和 dfs 部分仅用于生成符合描述的随机数据，实际使用时可忽略。
# 这里模拟了多个DataFrame，每个DataFrame代表一个'identifier'的数据块。
rng = np.random.default_rng(seed=42)
dfs = {
    idname: pd.DataFrame(data=[
        {
            'Labels': string.ascii_uppercase[i] + str(i + 1),
            'Numbers': rng.integers(0, 1000)
        } for i in range(20) # 每个DataFrame包含20个标签和数值
    ]) for idname in ['identifier1', 'identifier2', 'identifier3']
}
# ------------------------------------------------------------------

# 定义您希望输出到文件的特定字段列表（即DataFrame中的'Labels'）
desired_fields = [string.ascii_uppercase[i] + str(i + 1) for i in range(11)]
# 定义每行输出的键值对数量（步长），例如每4个键值对换行
stride = 4

# 初始化最终的输出字符串
outstr = ''

# 1. 添加文件头部注释
outstr += '// comment1
// comment2
// comment3
// comment4
'

# 2. 遍历每个标识符及其对应的数据块
for idname, id_data in dfs.items():
    # 2.1. 添加数据块的头部信息
    # 注意：原始格式中的 'label2 = i  label3' 包含反斜杠，需要转义
    outstr += f'{idname}       label2 = i \ label3        label4
label5
'

    # 2.2. 遍历所需的字段并提取、格式化数据
    for i, field in enumerate(desired_fields):
        # 从当前DataFrame中查找匹配的标签并获取对应的数值
        try:
            # 使用loc结合布尔索引查找，确保精确匹配特定行
            value = str(id_data.loc[id_data['Labels'] == field, 'Numbers'].iloc[0])
        except IndexError:
            # 处理数据缺失的情况：如果某个desired_field在当前DataFrame中不存在
            value = 'N/A' # 可以设置为'N/A'或任何默认值
            print(f"警告: 在 {idname} 中未找到标签 '{field}'，将使用 '{value}'。")

        # 格式化键值对，例如 "A1 = 89 "
        outstr += f'{field} = {value} '

        # 根据步长判断是否需要添加换行符
        if i % stride == stride - 1:
            outstr += '
'

    # 2.3. 如果最后一个字段没有以换行符结束（即最后一行不满stride），则添加一个换行符
    if not outstr.endswith('
'):
        outstr += '
'

    # 2.4. 在不同数据块之间添加一个空行作为分隔
    outstr += '
'

# 3. 将构建好的字符串打印到控制台并写入文件
print("--- 生成的输出内容预览 ---")
print(outstr)

output_filename = 'outputfile.txt'
with open(output_filename, 'w', encoding='utf-8') as fh:
    fh.write(outstr)
print(f"
内容已成功写入到文件: {output_filename}")

// comment1
// comment2
// comment3
// comment4
identifier1       label2 = i  label3        label4
label5
A1 = 89 B2 = 773 C3 = 654 D4 = 438 
E5 = 433 F6 = 858 G7 = 85 H8 = 697 
I9 = 201 J10 = 94 K11 = 526 

identifier2       label2 = i  label3        label4
label5
A1 = 500 B2 = 370 C3 = 182 D4 = 926 
E5 = 781 F6 = 643 G7 = 402 H8 = 822 
I9 = 545 J10 = 443 K11 = 450 

identifier3       label2 = i  label3        label4
label5
A1 = 165 B2 = 758 C3 = 700 D4 = 354 
E5 = 67 F6 = 970 G7 = 445 H8 = 893 
I9 = 677 J10 = 778 K11 = 759