本文主要介绍Python中,使用NumPy时,将其中的array数组存储到csv文件的方法,以及相关的示例代码。

1、使用numpy的savetxt()方法

相关文档http://docs.scipy.org/doc/numpy/reference/generated/numpy.savetxt.html

import numpy
np_array = numpy.asarray([ [1,2,3], [4,5,6], [7,8,9] ])
numpy.savetxt("foo.csv", np_array, delimiter=",")

2、使用pandas中DataFrame的to_csv()方法

import pandas as pd 
import numpy
np_array = numpy.asarray([ [1,2,3], [4,5,6], [7,8,9] ])
pd.DataFrame(np_array).to_csv("path/to/file.csv")

3、使用numpy的tofile()方法

相关文档http://docs.scipy.org/doc/numpy/reference/generated/numpy.ndarray.tofile.html

import numpy as np
np_array = np.asarray([ [1,2,3], [4,5,6], [7,8,9] ])
np_array.tofile('foo.csv',sep=',',format='%10.5f')

4、使用csv实现

相关文档https://docs.python.org/3/library/csv.html

import csv
import numpy
np_array = numpy.asarray([ [1,2,3], [4,5,6], [7,8,9] ])
with open('out2.csv', 'w', newline='') as fp:
    writer = csv.writer(fp, quoting=csv.QUOTE_NONNUMERIC)
    #writer.writerow(np_array.dtype.names)
    writer.writerows(np_array.tolist())

5、安装NumPy

pip install numpy

如果在Mac或Linux中,安装numpy速度如果比较慢,可以指定国内源,使用如下命令下载:

pip3 install numpy -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com


推荐文档