如何为字典对应键累加值？碱基位置统计代码问题求助

你遇到的核心问题确实是新值覆盖旧值——原来的代码每次找到碱基时，都会直接给对应位置赋值一个新的单元素集合，完全替换掉之前的内容。要实现同一个位置的碱基累加，我们需要先判断该位置是否已经在字典中，再决定是初始化集合还是往已有集合里添加元素。

这里有两种简洁的修改方式：

方法1：使用字典的setdefault方法 #

setdefault会先检查键是否存在：如果不存在，就用指定的默认值创建这个键；如果存在，就返回已有的值。这样我们可以直接往返回的集合里添加碱基：

def chunks(chunks_file):
    set_bases = {}
    with open(chunks_file) as file:
        for line in file:
            for character in line:
                if character.isalpha():
                    letter = character
                    position = line.find(letter) + 1
                    # 先获取或初始化该位置的集合，再添加当前碱基
                    set_bases.setdefault(position, set()).add(letter)
    return set_bases

方法2：显式判断键是否存在 #

如果你觉得setdefault不够直观，也可以用条件判断来实现同样的逻辑：

def chunks(chunks_file):
    set_bases = {}
    with open(chunks_file) as file:
        for line in file:
            for character in line:
                if character.isalpha():
                    letter = character
                    position = line.find(letter) + 1
                    if position in set_bases:
                        # 位置已存在，直接添加碱基到集合
                        set_bases[position].add(letter)
                    else:
                        # 位置不存在，初始化一个包含当前碱基的集合
                        set_bases[position] = {letter}
    return set_bases

额外注意 #

需要提醒的是：line.find(letter)会返回第一个匹配到该字符的索引，如果你的文本中同一个索引位置有多个相同碱基，这个逻辑没问题，但如果是不同索引位置有相同碱基，需要确认position的计算逻辑是否和你预期的“第1位、第2位”完全匹配。不过从你的问题描述来看，只要解决集合累加的问题，就能得到你想要的输出结果了。

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