如何用Numpy实现循环计算双列数据并覆写指定行而非追加？

原代码的问题在于：seek(j)是按字节偏移定位，而非行号，直接追加或随机写入会破坏文件结构，无法精准覆写指定行。正确的思路是先读取文件全部内容，更新目标行后再整体覆盖写入，以下是基于Numpy的实现：

import numpy as np

n = 3
output_file = 'test.txt'

# 生成需要更新的j值及对应计算结果
target_j = np.arange(0, n + 1)
target_y = 12 * np.sqrt(target_j)

try:
    # 读取现有文件数据，按两列浮点型加载
    existing_data = np.loadtxt(output_file, dtype=np.float64)
    # 检查现有数据的最大j值，若小于当前n则补充缺失行
    max_existing_j = int(existing_data[-1, 0]) if existing_data.size > 0 else -1
    if max_existing_j < n:
        new_j = np.arange(max_existing_j + 1, n + 1)
        new_y = 12 * np.sqrt(new_j)
        existing_data = np.vstack((existing_data, np.column_stack((new_j, new_y))))
    # 遍历目标j，更新对应行的第二列
    for j, y_val in zip(target_j, target_y):
        row_idx = np.where(existing_data[:, 0] == j)[0][0]
        existing_data[row_idx, 1] = y_val
except FileNotFoundError:
    # 文件不存在时，直接生成全量数据
    existing_data = np.column_stack((target_j, target_y))

# 覆盖写入文件，指定格式保证第一列为整数，第二列保留精度
np.savetxt(output_file, existing_data, fmt='%d %.15f')

关键说明 #

• 读取文件时用np.loadtxt解析现有数据，自动处理列结构；
• 若文件已存在且包含更大的j值，仅更新指定j行的第二列，其余行完全保留；
• 若现有数据的j值范围小于当前n，自动补充缺失的行；
• 用np.savetxt覆盖写入，通过fmt参数控制输出格式，匹配期望的结果样式。

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