如何用Python提取指定行并生成含XYZ坐标的列格式文件

需求说明 #

从大型数据文件中提取以ABC 开头的行，将每行的XYZ坐标提取后生成新文件，每行格式为X值, Y值, Z值，同时可保留原代码的求和统计逻辑，方便后续计算标准差。

修改后的代码实现 #

# 同时打开输入文件和输出文件，自动管理资源
with open("file", 'r') as f_in, open("output_coords.csv", 'w') as f_out:
    sums = [0] * 3
    points = 0
    for line in f_in:
        # 精准匹配以"ABC "开头的行（带空格避免误匹配类似ABCxxx的内容）
        if line.startswith("ABC "):
            points += 1
            # 分割行内容，提取第1-3位的坐标值
            coords = line.split()[1:4]
            # 将坐标转为逗号分隔格式并写入输出文件
            f_out.write(', '.join(coords) + '\n')
            # 保留原有的求和逻辑
            for i, x in enumerate(coords):
                sums[i] += float(x)

# 可选：输出提取统计信息
print(f"共提取{points}个有效坐标点")
print(f"X/Y/Z坐标总和：{sums[0]}, {sums[1]}, {sums[2]}")

关键细节说明 #

• 使用with语句同时管理输入输出文件，无需手动关闭，处理大型文件时内存占用更合理
• line.split()[1:4]确保只提取XYZ三个坐标，避免行内存在多余字段的干扰
• ', '.join(coords)直接将坐标列表转为目标格式，简洁高效

扩展：直接计算标准差 #

如果需要直接计算各列的标准差，可在提取时记录每个坐标值，结合统计公式计算：

import math

with open("file", 'r') as f_in, open("output_coords.csv", 'w') as f_out:
    sums = [0] * 3
    square_sums = [0] * 3
    points = 0
    for line in f_in:
        if line.startswith("ABC "):
            points += 1
            coords = line.split()[1:4]
            f_out.write(', '.join(coords) + '\n')
            x, y, z = map(float, coords)
            sums[0] += x
            sums[1] += y
            sums[2] += z
            square_sums[0] += x**2
            square_sums[1] += y**2
            square_sums[2] += z**2

# 计算总体标准差
if points > 0:
    std_x = math.sqrt((square_sums[0] - (sums[0]**2)/points) / points)
    std_y = math.sqrt((square_sums[1] - (sums[1]**2)/points) / points)
    std_z = math.sqrt((square_sums[2] - (sums[2]**2)/points) / points)
    print(f"X列标准差：{std_x:.6f}")
    print(f"Y列标准差：{std_y:.6f}")
    print(f"Z列标准差：{std_z:.6f}")
else:
    print("未找到符合条件的行")

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