Python教程

python : 将txt文件中的数据读为numpy数组或列表

本文主要是介绍python : 将txt文件中的数据读为numpy数组或列表,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

很多时候,我们将数据存在txt或者csv格式的文件里,最后再用python读取出来,存到数组或者列表里,再做相应计算。本文首先介绍写入txt的方法,再根据不同的需求(存为数组还是list),介绍从txt读取浮点数的方法。

一、写入浮点数到txt文件:

假设每次有两个浮点数需要写入txt文件,这里提供用with关键字打开文件的方法,使用with打开文件是一个很好的习惯,因为with结束,它就会自动close file,不用手动再去flie.close()。

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数组,那么一句代码可以搞定:

a = numpy.loadtxt('odom.txt')
存为numpy数组以后,a的尺寸可以用a.shape进行查看,这里应为:(14,2)也就是14行2列。如果你觉得这个形状不是你期待的,你可以用:

b = numpy.reshape(a,(你想要的形状))

b = numpy.reshape(a,(-1,1,2))
那么b就是一个三维数组了(14,1,2)
2.如果只是想读出来,存到一个list里,那就用下面的常用python格式了:

with open('odom.txt', 'r') as f:
data = f.readlines() #txt中所有字符串读入data

for line in data:
odom = line.split() #将单个数据分隔开存好
numbers_float = map(float, odom) #转化为浮点数
print numbers_float

 

这篇关于python : 将txt文件中的数据读为numpy数组或列表的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!