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="# " 。