这篇文章解释了如何将 NumPy 数组写入 CSV 文件。
我们将看:
将不同的 NumPy 数组写入 CSV 的语法
将 NumPy 数组写入 CSV 的限制
保存 NumPy 数组的替代方法
让我们开始吧。
将 NumPy 数组写入 CSV
您可以使用该np.savetxt()方法来保存…
确保:
将“.csv”添加到文件名目的地,然后
将分隔符关键字设置为“,”
如果您不使用这两个设置,NumPy 会将您的文件保存为 .txt。稍后再谈。
CSV 文件可能很棒,因为它们是人类可读的。它们还具有易于加载到 Pandas 或 Dask DataFrames 的额外好处。
写一维数组
让我们使用 来创建一个包含随机数的一维数组np.random.rand()。
import numpy as np
# 创建一维数组
a = np.array([1,2,3])
# 存储在当前目录
np.savetxt( "a.csv", a, delimiter="," )
要改为按行写入数据,请将newlinekwarg设置为“,”(您的分隔符)。
按行写入数组
np.savetxt("very.csv", a, delimiter=",", newline=",")
您还可以通过先将数组转换为二维数组来逐行写入数组
np.savetxt("very.csv", [a], delimiter="," )
在这两种情况下,的内容very.csv将如下所示:
写二维数组
现在让我们创建一个二维 NumPy 数组并将其保存为 CSV。
# 创建二维数组
b = np.array([1, 2, 3], [4, 5, 6])
# 将二维数组写入 CSV
np.savetxt( "merry.csv", b, delimiter="," )
如您所料,默认情况下二维数组按行写入。
内容merry.csv:
1 2 3
4 5 6