如何使用Python将含重复'A'标识的字典键值按指定规则写入CSV文件

我来帮你搞定这个需求！首先得明确核心目标：从text1里提取第一个"A"到下一个"A"之前的所有元素作为表头，再把text2的数据按表头长度分组，逐行匹配写入CSV。

先看看你原代码的问题：它只是逐个输出text1的元素，遇到后续的"A"就跳过，但既没正确提取表头，也没处理text2的对应数据，最终输出肯定不符合预期。

修正后的实现代码 #

import csv

data = {
    "text1": ("A", "mouse", "cat", "A", "mouse", "cat", "A", "mouse", "cat"),
    "text2": (1, 2, 3, 4, 5, 6, 7, 8, 9)
}

# 第一步：提取表头——从第一个"A"到下一个"A"之前的所有元素
text1_list = list(data["text1"])
# 找到第一个"A"的索引
first_a_idx = text1_list.index("A")
# 找到第二个"A"的索引（从第一个A的下一位开始找）
second_a_idx = text1_list.index("A", first_a_idx + 1)
# 提取表头：第一个A到第二个A之前的所有元素
header = text1_list[first_a_idx:second_a_idx]

# 第二步：计算每组数据的长度（和表头长度一致）
group_length = len(header)
# 按group_length对text2进行分组
text2_groups = [data["text2"][i:i+group_length] for i in range(0, len(data["text2"]), group_length)]

# 第三步：写入CSV文件
with open("result.csv", "w", newline="") as f:
    writer = csv.writer(f, delimiter=",")
    # 写入表头
    writer.writerow(header)
    # 逐行写入分组后的text2数据
    for group in text2_groups:
        writer.writerow(group)

# 验证输出
with open("result.csv", "r") as f:
    print(f.read())

代码解释 #

1. 提取表头：

   • 先把tuple类型的text1转成list方便操作
   • 用index()方法找到第一个和第二个"A"的位置，截取中间的元素作为表头（示例中表头是["A", "mouse", "cat"]）

2. 分组text2数据：

   • 表头的长度就是每组数据的长度（示例中是3）
   • 用列表推导式按这个长度对text2进行切片，得到[(1,2,3), (4,5,6), (7,8,9)]这样的分组

3. 写入CSV：

   • 先写表头行，再把每个分组作为一行写入，确保每列数据和表头一一对应

最终输出的CSV内容 #

A,mouse,cat
1,2,3
4,5,6
7,8,9

这样就完全符合你的需求啦！

内容的提问来源于stack exchange，提问作者Hoang Minh Nguyen