日志

Python科学计算技巧积累二——读取txt文件

热度 1已有 2691 次阅读| 2015-5-8 12:45 |个人分类:编程语言

读取数据是后期数据处理的必要步骤。.txt是广泛使用的数据文件格式。一些.csv, .xlsx等文件可以转换为.txt 文件进行读取。我常使用的是Python自带的I/O接口，将数据读取进来存放在list中，然后再用numpy科学计算包将list的数据转换为array格式，从而可以像MATLAB一样进行科学计算。

下面是一段常用的读取txt文件代码，可以用在大多数的txt文件读取中
filename = 'array_reflection_2D_TM_vertical_normE_center.txt' # txt文件和当前脚本在同一目录下，所以不用写具体路径

pos = []

Efield = []

with open(filename, 'r') as file_to_read:

while True:

        lines = file_to_read.readline() # 整行读取数据
        if not lines:
            break
            pass

p_tmp, E_tmp = [float(i) for i in lines.split()] # 将整行数据分割处理，如果分割符是空格，括号里就不用传入参数，如果是逗号，则传入‘，’字符。

pos.append(p_tmp) # 添加新读取的数据

Efield.append(E_tmp)

pass

pos = np.array(pos) # 将数据从list类型转换为array类型。

Efield = np.array(Efield)

pass