Python NumPy savetxt方法用法及代码示例

例子

基本用法

保存一维 Numpy 整数数组：

x = np.array([3,4,5])
np.savetxt("my_data", x)

这将在与 Python 脚本相同的目录中创建名为 "my_data" 的以下文本文件：

3.000000000000000000e+00
4.000000000000000000e+00
5.000000000000000000e+00

正如上面 fmt 参数中所解释的，所有值都存储为类型 float，精度为 18 - 即 18 个零！

由于在本例中这不是您想要的，因此您可以指定 fmt 参数，如下所示：

x = np.array([3,4,5])
np.savetxt("my_data", x, fmt="%d")

这里，%d只是意味着将值存储为整数。检查我们的"my_data"：

3
4
5

我们看到数字确实存储为整数。

指定分隔符参数

用于分隔值的默认分隔符是单个空格。我们可以使用 delimiter 参数指定一个：

x = np.array([[3,4],[5,6]])
np.savetxt("my_data", x, fmt="%d", delimiter=",")

我们最终得到以下 "my_data" 文本文件：

3,4
5,6

请注意，分隔符仅在数组是二维时才起作用。

指定换行参数

默认情况下，对于一维数组，每个值都存储在单独的行中，对于二维数组，每行都存储在单独的行中。

要使用 @ 来分隔行：

x = np.array([[3,4],[5,6]])
np.savetxt("my_data", x, fmt="%d", newline="@")

输出如下：

3 4@5 6@

指定标题

要保存带有标头的一维整数数组：

x = np.array([3,4,5])
np.savetxt("my_data", x, fmt="%d", header="My numbers")

我们的"my_data"文件如下：

# My numbers
3
4
5

请注意# 是如何附加到我们指定的标头中的。您可以使用comments 参数来控制它。

指定页脚

要保存带有页脚的一维整数数组：

x = np.array([3,4,5])
np.savetxt("my_data", x, fmt="%d", footer="My footer")

我们的"my_data"文件如下：

3
4
5
# My footer

再次注意 # 是如何附加到我们指定的页脚的。您可以使用comments 参数来控制它。

指定注释

要保存标头附加前缀 "// " 的一维整数数组：

x = np.array([3,4,5])
np.savetxt("my_data", x, fmt="%d", header="My numbers", comments="// ")

输出如下：

// My numbers
3
4
5

请注意，仅当设置了 header 或 footer 时，comments 才相关。另外，默认情况下为 comments="# " 。