Pythonic 方式替换字符

Question

我想使用 pythonic 方法替换字符串中的一些字符。

A -> T
C -> G
G -> C
T -> A

例子：

AAATCGATGAT

将转变为

TTTAGCTAACTA

我做了什么：

def swap(string):
    string = re.sub('A', 'aux', string)
    string = re.sub('T', 'A', string)
    string = re.sub('aux', 'T', string)
    string = re.sub('C', 'aux', string)
    string = re.sub('G', 'C', string)
    string = re.sub('aux', 'G', string)

    return string

它效果很好，但我正在寻找一种更Pythonic的方式来实现这一目标。

Answer 1

使用带有理解式和 的字典str.join：

translateDict = {
  "A" : "T",
  "C" : "G",
  "G" : "C",
  "T" : "A"
}

s1 = "AAATCGATTGAT"
reconstructed = "".join(translateDict.get(s, s) for s in s1)

这里有一个生动的例子

注意使用dict.get; 如果字典中没有这封信，我们就让它保持原样。

正如 @bravosierra99 所建议的，您也可以简单地使用str.translate：

reconstructed = s1.translate(string.maketrans(translateDict))