如何用python将csv文件中的数据读取成数组

Question

如题所述

Answer 1

很多时候，我们将数据存在txt或者csv格式的文件里，最后再用python读取出来，存到数组或者列表里，再做相应计算。本文首先介绍写入txt的方法，再根据不同的需求（存为数组还是list）,介绍从txt读取浮点数的方法。
一、写入浮点数到txt文件：
假设每次有两个浮点数需要写入txt文件，这里提供用with关键字打开文件的方法，使用with打开文件是一个很好的习惯，因为with结束，它就会自动close file，不用手动再去flie.close()。

[python] view plain copy
with open('file_path/filename.txt','a') as file:
write_str = '%f %f\n'%(float_data1,float_data2)
file.write(write_str)

二、从txt文件提取浮点数到numpy数组或list列表

首先假设你有一个odom.txt 的文件，里面每一行的数据个数都相同，如截图所示：

1.如果你是想把这些数据读取出来，存为numpy数组,那么一句代码可以搞定：

[python] view plain copy
a = numpy.loadtxt('odom.txt')

存为numpy数组以后，a的尺寸可以用a.shape进行查看，这里应为：（14,2）也就是14行2列。如果你觉得这个形状不是你期待的，你可以用:

[python] view plain copy