如何通过Python修改data.py文件中的accidents列表并添加新元素？

你之前的代码只是修改了内存中的accidents列表对象，不会同步到源文件里——这就是为什么函数内打印有变化，但全局打印和文件本身都没改动的原因。要真正修改data.py，得直接操作文件内容，下面给两种可行的方法：

方法1：正则替换（适合单行简单列表） #

针对你给出的accidents=[a, b, c]这种单行列表场景，用正则匹配替换最直接：

import re

# 读取data.py内容
with open('data.py', 'r', encoding='utf-8') as f:
    content = f.read()

# 要添加的新元素（如果是字符串，改成"'d'"）
new_item = 'd'

# 匹配accidents列表定义行
pattern = r'(accidents\s*=\s*\[)(.*?)(\])'
match_result = re.search(pattern, content)
if match_result:
    prefix, existing_items, suffix = match_result.groups()
    # 处理逗号分隔逻辑
    if existing_items.strip():
        updated_items = f'{existing_items}, {new_item}'
    else:
        updated_items = new_item
    # 替换并写回文件
    updated_content = re.sub(pattern, f'{prefix}{updated_items}{suffix}', content)
    with open('data.py', 'w', encoding='utf-8') as f:
        f.write(updated_content)

运行后，data.py里的accidents=[a, b, c]会变成accidents=[a, b, c, d]。

方法2：AST解析（兼容复杂列表结构） #

如果你的accidents列表是多行、带注释或者有其他复杂格式，正则容易出错，用AST（抽象语法树）解析代码结构更可靠：

import ast
import astunparse

# 读取并解析data.py为AST树
with open('data.py', 'r', encoding='utf-8') as f:
    tree = ast.parse(f.read())

# 构造要添加的元素节点（如果是字符串，改成ast.Constant(value='d')）
new_element = ast.Name(id='d', ctx=ast.Load())

# 遍历AST找到accidents的赋值语句并修改
for node in ast.walk(tree):
    if (isinstance(node, ast.Assign)
        and len(node.targets) == 1
        and isinstance(node.targets[0], ast.Name)
        and node.targets[0].id == 'accidents'):
        if isinstance(node.value, ast.List):
            node.value.elements.append(new_element)
            break

# 将修改后的AST转换回代码文本
updated_content = astunparse.unparse(tree).strip()

# 写回文件
with open('data.py', 'w', encoding='utf-8') as f:
    f.write(updated_content)

注意：需要先安装astunparse库，执行pip install astunparse即可。这种方法能正确处理各种合法的Python列表格式，不会破坏原文件的代码结构。

注意事项 #

• 操作文件前建议先备份data.py，避免意外导致代码丢失。
• 如果要添加的是字符串常量（比如'd'），记得调整代码里的新元素定义，避免语法错误。

内容的提问来源于stack exchange，提问作者G'ayrat Raxmadullayev