如何将包含x,y,z坐标的嵌套列表写入带原子名前缀的3列xyz文件

原有代码问题梳理 #

• 外层循环生成了3个xyz文件，若只需单个ring1.xyz无需额外外层循环
• 内层循环遍历坐标子列表的单个数值，导致每行仅输出单个坐标值，不符合一行放原子名+三个坐标的格式要求
• 没有读取y列表的坐标数据写入文件
• 写入时多余插入了换行符，导致H和坐标拆分到不同行

修正后代码 #

场景1：生成单个ring1.xyz，匹配你给出的预期输出格式 #

# 定义坐标列表
x = [[1,2,3],[4,5,6],[7,8,9]]
y = [[10,11,12],[13,14,15],[16,17,18]]

# 写入文件
with open("ring1.xyz", "w") as f:
    # 写入x的前两个子列表
    for coord in x[:2]:
        f.write(f"H {coord[0]} {coord[1]} {coord[2]}\n")
    # 写入y的前两个子列表
    for coord in y[:2]:
        f.write(f"H {coord[0]} {coord[1]} {coord[2]}\n")

场景2：保留原代码逻辑生成3个对应下标的xyz文件 #

如果你的需求是生成ring_0.xyz、ring_1.xyz、ring_2.xyz三个文件，每个文件包含对应下标的x、y坐标，可以用以下代码：

x = [[1,2,3],[4,5,6],[7,8,9]]
y = [[10,11,12],[13,14,15],[16,17,18]]

for s in range(3):
    with open(f"ring_{s}.xyz", "w") as f:
        f.write(f"H {x[s][0]} {x[s][1]} {x[s][2]}\n")
        f.write(f"H {y[s][0]} {y[s][1]} {y[s][2]}\n")

代码说明 #

• 用列表切片[:2]直接取每个嵌套列表的前两个子列表，无需手动控制下标范围
• 用f-string格式化每行内容，直接将原子名和三个坐标拼接为同一行，自动添加换行符
• 打开文件用w模式即可，无需w+（读写模式），仅写入操作不需要读权限

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