如何使用numpy在单个输出文件中连续存储多组数据及文本？

嘿，这个需求其实很容易实现，结合numpy的savetxt和Python基础的文件操作就能搞定！我给你两种实用的方案，你可以根据自己的习惯选：

方法一：分步骤追加写入 #

这种方式适合需要分开处理不同内容的场景，先写矩阵A，再追加文本，最后追加矩阵B：

import numpy as np

# 先定义示例矩阵（换成你自己的矩阵就行）
matrix_a = np.array([[1, 2, 3], [4, 5, 6]])
matrix_b = np.array([[7, 8], [9, 10], [11, 12]])

# 1. 写入矩阵A到文件（'w'模式会创建/覆盖文件）
np.savetxt('output.txt', matrix_a, fmt='%d')

# 2. 追加指定文本到文件末尾
with open('output.txt', 'a') as f:
    f.write('Matrix A "and below that some text and the another" matrix B\n')

# 3. 追加矩阵B到文件末尾（注意用'a'模式）
np.savetxt('output.txt', matrix_b, fmt='%d', header='', footer='')

方法二：一次性打开文件批量写入 #

如果不想多次打开关闭文件，更高效的方式是一次性打开文件，依次写入所有内容：

import numpy as np

matrix_a = np.array([[1, 2, 3], [4, 5, 6]])
matrix_b = np.array([[7, 8], [9, 10], [11, 12]])

with open('output.txt', 'w') as f:
    # 先写矩阵A
    np.savetxt(f, matrix_a, fmt='%d')
    # 再写指定文本
    f.write('Matrix A "and below that some text and the another" matrix B\n')
    # 最后写矩阵B
    np.savetxt(f, matrix_b, fmt='%d', header='', footer='')

关键注意点： #

• fmt参数：用来控制数据的输出格式，比如%d是整数格式，%.2f是保留两位小数的浮点数，根据你的矩阵数据类型调整。
• 文件模式：'w'会覆盖已有文件，'a'是追加模式；如果是第一次生成文件，用'w'更稳妥。
• 去掉默认注释：savetxt默认会在开头加#注释行，设置header=''和footer=''可以避免这个问题。

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