NumPy的文件读写
NumPy中使用np.loadtxt()或者更加专门化的np.genfromtxt()将数据加载到普通的Numpy数组中,savetxt() 将数据保存到磁盘文件里。这些函数都有许多选项可供使用:指定各种分隔符、跳过行数等。我们以一个简单的逗号分割文件(csv)为例:
import numpy as np
ndarray1 = np.array([
['aaa', 'bbb', 'ccc', 'ddd', 'eee'],
['fff', 'ggg', 'hhh', 'iii', 'jjj'],
['aaa', 'bbb', 'ccc', 'ddd', 'eee'],
['fff', 'ggg', 'hhh', 'iii', 'jjj']
])
# 将数组保存成csv文件,每个数据之间用逗号隔开
np.savetxt('ndarray1.csv', ndarray1, delimiter=',', fmt='%s')
np.genfromtxt('ndarray1.csv', delimiter=',', dtype=np.unicode)
# array([['aaa', 'bbb', 'ccc', 'ddd', 'eee'],
# ['fff', 'ggg', 'hhh', 'iii', 'jjj'],
# ['aaa', 'bbb', 'ccc', 'ddd', 'eee'],
# ['fff', 'ggg', 'hhh', 'iii', 'jjj']],
# dtype='<U3')