利用pandas对txt文件进行数据处理

记一次帮朋友进行txt数据处理问题。
需求:对大量txt文件中第一行的数据进行处理。将第二个数据与第四个数据相加,并将相加结果替换到第四个数据位置,将第三个数据与第五个数据相加,并将相加结果替换到第五个数据位置,其余数据不变。
通过对问题的搜索与查询,得出以下代码:

import os
from glob import glob
import pandas as pd

#输入需要批处理的txt文件路径
paths = glob(r'*.txt')
for path in paths:
    file = os.path.basename(path)  #返回文件名
    #print(file)
    dates = pd.read_csv(file,header=None,sep = ' ')#sep:每个数直接是用空格隔开的
    #print(dates[1][0])#1:列数 0:行数
    #print(dates[2][0])
    #print(dates[3][0])
    #print(dates[4][0])
    data1=float(dates[1][0])+float(dates[3][0])
    data2 = float(dates[2][0]) + float(dates[4][0])
    #print(data1)
    #print(data2)
    with open(file, "w+") as f:
        f.write(str(dates[0][0])+' '+str(dates[1][0])+' '+str(dates[2][0])+' '+str(data1)+' '+str(data2))  

你可能感兴趣的:(python,python)