numpy.savetxt详解

numpy.savetxt详解

Numpy是一个Python中常用的库，用于进行科学计算。其中的numpy.savetxt函数可以将数组保存到文本文件中。本文将详细解释numpy.savetxt函数的用法，包括必要的参数和示例代码。

语法

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

参数说明：

• fname: 字符串，要保存的文件或文件对象。
• X: 数组，要保存的数据。
• fmt: 字符串格式控制，用于指示如何保存数据，默认值为’%.18e’。
• delimiter: 字符串，用于分隔数据的字符串，默认为一个空格。
• newline: 字符串，用于换行的字符串，默认为换行符。
• header: 字符串，要写在文件头部的字符串。
• footer: 字符串，要写在文件尾部的字符串。
• comments: 字符串，用于指示注释的前缀，默认为’#’。
• encoding: 字符串，指定文件编码的字符串，默认为None。

示例

示例1：保存一维数组

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
np.savetxt('array.txt', arr)

运行结果：在当前目录下生成一个名为array.txt的文件，内容为：

1.000000000000000000e+00
2.000000000000000000e+00
3.000000000000000000e+00
4.000000000000000000e+00
5.000000000000000000e+00

示例2：保存二维数组

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
np.savetxt('matrix.txt', arr, delimiter=',')

运行结果：在当前目录下生成一个名为matrix.txt的文件，内容为：

1.000000000000000000e+00,2.000000000000000000e+00,3.000000000000000000e+00
4.000000000000000000e+00,5.000000000000000000e+00,6.000000000000000000e+00

示例3：保存带有注释的数组

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
np.savetxt('matrix.txt', arr, delimiter=',', header='This is a matrix', comments='')

运行结果：在当前目录下生成一个名为matrix.txt的文件，内容为：

This is a matrix
1.000000000000000000e+00,2.000000000000000000e+00,3.000000000000000000e+00
4.000000000000000000e+00,5.000000000000000000e+00,6.000000000000000000e+00

总结

numpy.savetxt函数是一个非常方便的函数，可以将数组保存到文本文件中。通过对参数的配置，可以实现不同格式的输出。