如何将NumPy数组导出到CSV文件（附实例）

你可以使用以下基本语法将NumPy数组导出到CSV文件：

import numpy as np

#define NumPy array
data = np.array([[1,2,3],[4,5,6],[7,8,9]])

#export array to CSV file
np.savetxt("my_data.csv", data, delimiter=",")

下面的例子展示了如何在实践中使用这种语法。

例1：导出NumPy数组到CSV中

下面的代码展示了如何将一个NumPy数组导出到CSV文件：

import numpy as np

#define NumPy array
data = np.array([[1,2,3], [4,5,6], [7,8,9], [10, 11, 12], [13, 14, 15]])

#export array to CSV file
np.savetxt("my_data.csv", data, delimiter=",")

如果我导航到CSV文件在我笔记本上的保存位置，我就可以查看数据了。

例2：用特定的格式将NumPy数组导出到CSV中

数字的默认格式是"%.18e" - 这显示18个零。然而，我们可以使用fmt参数来指定不同的格式。

例如，下面的代码将一个NumPy数组导出到CSV并指定了两个小数位：

import numpy as np

#define NumPy array
data = np.array([[1,2,3], [4,5,6], [7,8,9], [10, 11, 12], [13, 14, 15]])

#export array to CSV file (using 2 decimal places)
np.savetxt("my_data.csv", data, delimiter=",", fmt="%.2f")

如果我导航到CSV文件的保存位置，我就可以查看数据了：

例子3：将NumPy数组导出到CSV中，并加上页眉

下面的代码展示了如何将一个NumPy数组导出到带有自定义列头的CSV文件：

import numpy as np

#define NumPy array
data = np.array([[1,2,3], [4,5,6], [7,8,9], [10, 11, 12], [13, 14, 15]])

#export array to CSV file (using 2 decimal places)
np.savetxt("my_data.csv", data, delimiter=",", fmt="%.2f",
           header="A, B, C", comments="")

注意：注释参数可以防止标题中显示 "#"符号。

如果我导航到CSV文件的保存位置，我就可以查看数据。

注意：你可以在这里找到numpy.savetxt()函数的完整文档。

其他资源

下面的教程解释了如何在Python中进行其他常见的读和写操作：

如何用NumPy读取CSV文件
如何用Pandas读取CSV文件
如何将Pandas DataFrame导出为CSV文件