如何用Python/Pandas将JSON字典转为带标签的分类任务TXT文件

实现思路 #

读取sample.json中的字典数据，遍历每个键值对：

• 若键以_0结尾，标记为label1
• 若键以_1结尾，标记为label2
• 将C语言代码内容与对应标签用制表符分隔，逐行写入sample.txt

Python实现代码 #

import json

# 加载JSON数据集
with open('sample.json', 'r', encoding='utf-8') as json_fp:
    c_code_data = json.load(json_fp)

# 生成分类任务TXT文件
with open('sample.txt', 'w', encoding='utf-8') as txt_fp:
    for item_key, code_content in c_code_data.items():
        # 判断标签
        if item_key.endswith('_0'):
            label = 'label1'
        elif item_key.endswith('_1'):
            label = 'label2'
        else:
            # 跳过不符合命名规则的条目，可根据需求调整
            continue
        # 写入制表符分隔的行
        txt_fp.write(f"{code_content}\t{label}\n")

注意事项 #

• 确保sample.json与脚本在同一目录，或修改文件路径为绝对路径
• 若JSON中的C代码包含换行符，写入TXT时会保留，不影响后续读取（按行分割样本即可）
• 可根据需求扩展逻辑，比如为不符合键规则的条目添加默认标签，或对代码内容做预处理

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