numpy.savetext 一行 - Python

numpy.savetext是一个函数，用于将数组写入文本文件。它可以方便地将数组保存为文本，并且可以对输出进行格式化。

语法

numpy.savetext(fname, X, fmt='%.18e', delimiter=' ', newline='\n', header='', footer='', comments='# ', encoding=None)

• fname: 文件名或文件描述符。如果是文件描述符，则需要设置 encoding=None。
• X: 数组。
• fmt: 输出格式。默认值是 %.18e。
• delimiter: 分隔符。默认值是空格。
• newline: 行分隔符。默认值是 \n。
• header: 文件头。
• footer: 文件尾。
• comments: 注释前缀。默认值是 # 。
• encoding: 编码格式。

示例代码

import numpy as np

# 创建数组
x = np.arange(10)
y = x**2

# 将数组保存为文本
np.savetxt('data.txt', np.column_stack((x, y)), fmt='%d', header='x,y', delimiter=',')

此代码将创建一个名为 data.txt 的文件。文件的内容类似于:

# x,y
0,0
1,1
2,4
3,9
4,16
5,25
6,36
7,49
8,64
9,81

总结

numpy.savetext是一个非常有用的函数。它可以方便地将数组保存为文本，并且具有很强的自定义能力。如果你需要将数组转换为文本文件，那么 numpy.savetext 绝对是一个很好的选择。