如何将 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 文件中。