将NumPy数组写入CSV文件中

最新推荐文章于 2024-09-22 15:15:19 发布

匆匆喂

最新推荐文章于 2024-09-22 15:15:19 发布

阅读量4.5k

点赞数 4

文章标签： numpy python pandas

本文链接：https://blog.csdn.net/liweiminlining/article/details/126365438

版权

本文介绍了如何使用NumPy的savetxt()和tofile()方法将数组保存到CSV文件，强调了避免使用不必要的数据转换方法以节省内存。 savetxt()允许指定分隔符和格式，而tofile()同样支持自定义分隔和格式。对于大规模数组，直接使用NumPy方法更为高效。

摘要由CSDN通过智能技术生成

使用numpy.savetxt()方法

当我们需要将NumPy数组转储到一个输出文件时，我们的第一个选择是numpy.savetxt()方法，该方法用于将数组保存到一个文本文件。该方法允许人们指定分隔符。
直接输出的将是科学符号，如果想避免使用科学符号，可以指定一个适当的fmt。

np.savetxt('output.csv', arr, delimiter=', ', fmt='%f')

使用numpy.tofile()方法

第二个选项是numpy.ndarray.tofile()方法，用于将数组写成文本或二进制文件。再一次，我们可以指定分隔符（sep）和格式。

arr.tofile('output.csv', sep=', ', format='%f')

要避免什么

另一种流行的（但不是很聪明的）方法是首先将NumPy数组转换为pandas DataFrame，然后使用pandas.DataFrame.to_csv()方法，以便将生成的DataFrame写入输出的csv文件中。
然而，请注意，这并不是一个好的做法，因为它将消耗太多的内存--尤其是当你在处理相当大的数组时--完全没有理由。