Python中在首个等号前插入&：两种方法的效率与风格对比

问题背景 #

需要处理包含多个等号的字符串，要求在首个等号（=）的前方插入&符号，示例输入字符串：

s = "x = y = z = 5"

方法1：列表转换+插入重组 #

通过将字符串转为列表，定位首个=的索引后插入&，最后用join方法重组字符串：

def replace_by_index(s):
    idx = s.index('=')
    s = list(s)
    s.insert(idx, '&')
    return "".join(s)

方法2：字符串切片拼接 #

利用字符串切片将原字符串在首个=处分割，直接拼接插入&后的新字符串：

def replace_at_index(s):
    idx = s.index('=')
    return s[:idx-1] + '&' + s[idx:]

注：原代码中的idx-1会将&插入到等号前的空格位置，若需求是在等号紧邻左侧插入（如将x =转为x &=），应改为idx。

效率对比 #

• 方法1：包含三次线性操作：字符串转列表（O(n)）、列表中间插入元素（O(n)，需移动后续元素）、列表转字符串（O(n)），整体时间复杂度为O(n)，但额外开销较多。
• 方法2：字符串切片和拼接都是Python底层优化的操作，总时间复杂度同样是O(n)，但省去了列表转换和元素移位的成本，实际运行效率更高，长字符串场景下差异更明显。

可读性讨论 #

两种方案的可读性因人而异：

• 方法1的逻辑更贴近自然语言表述，每一步操作意图清晰直白，对Python新手友好，容易理解完整流程。
• 方法2代码简洁紧凑，熟悉Python字符串特性的开发者能快速捕捉到“指定位置插入字符”的核心逻辑，显得更优雅。

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