Python写入CSV时小数位数过多的优化方案问询

核心解决方案：格式化每个浮点数为指定小数位数 #

你可以通过在写入前对每个坐标值进行四舍五入并格式化为三位小数的字符串，解决小数位数过多的问题。修改后的代码如下：

import csv

with open(filename, "w") as f:
    csvw = csv.writer(f, delimiter=";", quotechar='"')
    for point in points:
        # 对每个坐标执行四舍五入并保留三位小数
        formatted_point = [f"{coord:.3f}" for coord in point]
        csvw.writerow(formatted_point)

关键细节说明 #

• 使用f"{coord:.3f}"格式说明符：该语法会自动对浮点数进行十进制四舍五入，并保留三位小数，完全符合你要求的优先四舍五入而非截断的规则。
• 格式效果示例：
  • 0.04471017781221634 → "0.045"（四舍五入后）
  • -0.052283996037127314 → "-0.052"
  • 0.0 → "0.000"（若希望更紧凑，可改用f"{coord:.3g}"，会自动去除末尾无效零，0.0将变为"0"）

百万级数据的优化方案 #

如果你的points是numpy.ndarray类型的三维数组，推荐使用向量化操作提升处理效率，避免Python循环的开销：

import numpy as np
import csv

# 假设points是shape为(n, 3)的numpy浮点数组
formatted_points = np.char.mod("%.3f", points)
with open(filename, "w") as f:
    csvw = csv.writer(f, delimiter=";", quotechar='"')
    csvw.writerows(formatted_points)

这种方式利用numpy的批量处理能力，在百万级数据场景下会比纯Python循环快数倍，同时保证输出格式符合要求。

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