将NumPy数组写入CSV文件中

使用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文件中。
然而,请注意,这并不是一个好的做法,因为它将消耗太多的内存--尤其是当你在处理相当大的数组时--完全没有理由。

你可能感兴趣的:(算法,大数据,工具,numpy,python,pandas)