如何将numpy数组存成.csv文件
这篇文章解释了如何将 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
上一篇:
通过多线程提高代码的执行效率例子