如何將 NumPy 陣列儲存到文字檔案?


NumPy 陣列可以透過多種方法儲存到文字檔案,例如 savetxt() 方法、save() 方法和 dataframe.to_csv() 函式。NumPy 是一個 Python 庫,用於進行數值計算、運算元組等。本文將討論可用於將 NumPy 陣列儲存到文字檔案的方法。

方法 1:使用 numpy.savetxt() 函式

numpy.savetxt() 函式簡單地將 NumPy 陣列儲存到文字檔案。該函式接受兩個引數 - 要將 NumPy 陣列儲存到的檔名以及陣列本身。陣列必須是二維陣列,並且檔案以純文字格式儲存,每個元素之間用分隔符隔開。

示例

在下面的示例中,我們建立了一個二維 NumPy 陣列 arr,並使用 numpy.savetxt() 函式將其儲存到名為“myarray.txt”的文字檔案中。文字檔案中使用的分隔符是空格,這是預設分隔符。您可以透過在 numpy.savetxt() 函式中指定“delimiter”引數來更改分隔符。

import numpy as np

# Create a 2D NumPy array
arr = np.array([[1, 2, 3], [4, 5, 6]])

# Save the array to a text file
np.savetxt('myarray.txt', arr)

輸出

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

方法 2:使用 numpy.save() 函式

numpy.save() 函式將陣列儲存到具有`.npy`副檔名的二進位制檔案中。可以使用 numpy.load() 函式將檔案載入回 Python。

示例

在下面的示例中,我們建立了一個 NumPy 陣列 arr,並使用 numpy.save() 函式將其儲存到名為“myarray.npy”的二進位制檔案中。生成的檔案是二進位制檔案,可以使用 numpy.load() 函式將其載入回 Python。

import numpy as np

# Create a NumPy array
arr = np.array([1, 2, 3, 4, 5])

# Save the array to a binary file
np.save('myarray.npy', arr)

# Load the saved array
arr_loaded = np.load('myarray.npy')
print(arr_loaded)

輸出

[1 2 3 4 5]

方法 3:使用 pandas.DataFrame.to_csv() 函式

dataframe.to_csv() 方法用於將 NumPy 陣列儲存到 CSV 檔案。為此,您需要使用 Pandas 庫將二維陣列轉換為 Dataframe,然後使用 to_csv() 函式儲存它。Pandas 在 CSV 檔案中提供了更多格式選項。

示例

在下面的示例中,我們首先建立一個二維 NumPy 陣列 arr。然後,我們使用 Pandas 將陣列轉換為 DataFrame,並將 DataFrame 儲存到名為“myarray.csv”的 CSV 檔案中。我們將 index 和 header 引數指定為 False 以將其從 CSV 檔案中排除。CSV 檔案中使用的分隔符是逗號,這是 to_csv() 函式的預設分隔符。您可以透過在函式中指定“sep”引數來更改分隔符。

import numpy as np
import pandas as pd

# Create a 2D NumPy array
arr = np.array([[1, 2, 3], [4, 5, 6]])

# Convert the array to a DataFrame
df = pd.DataFrame(arr)

# Save the DataFrame to a CSV file
df.to_csv('myarray.csv', index=False, header=False)

輸出

1,2,3
4,5,6

應用

將 NumPy 陣列儲存到文字檔案是科學計算和資料分析中的一項常見任務。以下是使用將 NumPy 陣列儲存到文字檔案的一些常見應用和場景:

  • 資料分析 - NumPy 陣列廣泛用於資料分析,用於儲存和操作大型資料集。將 NumPy 陣列儲存到文字檔案使您可以輕鬆地與他人共享資料或將資料載入到其他程式語言或應用程式中。

  • 機器學習 - 機器學習演算法通常需要大型資料集進行訓練。將 NumPy 陣列儲存到文字檔案是儲存這些資料集並在未來的實驗或分析中重複使用它們的一種方法。

  • 訊號處理 - NumPy 陣列通常用於訊號處理應用程式中,用於儲存和操作音訊或影像資料。將 NumPy 陣列儲存到文字檔案允許將資料輕鬆載入到其他應用程式或離線分析。

  • 數值模擬 - NumPy 陣列廣泛用於科學模擬中,用於求解微分方程、求解線性系統和其他數值計算。將 NumPy 陣列儲存到文字檔案允許輕鬆儲存和分析模擬資料。

  • 實驗資料 - NumPy 陣列可用於儲存來自科學實驗或測量的實驗資料。將 NumPy 陣列儲存到文字檔案是將資料儲存為易於共享或稍後分析的格式的一種方法。

結論

本文討論瞭如何使用 numpy.savetxt()、numpy.save() 和 pandas.dataframe.to_csv() 等方法將 NumPy 陣列儲存到文字檔案。savetxt() 函式簡單地將 NumPy 陣列儲存到文字檔案。save() 方法將 NumPy 陣列儲存到可載入回 Python 的二進位制檔案中。to_csv() 方法將 NumPy 陣列儲存到 CSV 檔案中。

更新時間: 2023年7月11日

21K+ 次瀏覽

開啟你的 職業生涯

透過完成課程獲得認證

立即開始
廣告

© . All rights reserved.