Python读写文件

读文件

Readlines（）方法

file=open(‘output/speed.txt’,‘r’)
lines=file.readlines()

python读数据时:多加一行空行，结果正确。加两个空行时，倒数第二行也会被读入。

代码示例：多行处理函数：

输入speed.txt:

1 gpu1 Speed 29.26
1 gpu0 Speed 47.15

输出numpy数组转换的batch_speed.csv：

import numpy as np

def getGPUnum(lines):
        gpunum=1
        str_batchsize=lines[0].split()[0]
        i=1

        if len(lines)==1:
                return 1

        while i<len(lines) and lines[i].split()[0]==str_batchsize:
                gpunum=gpunum+1
                i=i+1
        return gpunum
if __name__ == '__main__':

        file=open('output/speed.txt','r')
        lines=file.readlines()

        gpunum=getGPUnum(lines)

        i=0
        print("Job has",gpunum,"gpu(s)")
        result=[]
        cur = []
        for line in lines:
                strs=line.split()
                if i%gpunum==0:
                        if i:
                                result.append(cur)
                                cur=[]
                        cur.append(int(strs[0]))
                        cur.append(float(strs[-1]))
                        # print(cur)
                else:
                        cur.append(float(strs[-1]))
                        # print(cur)
                i=i+1
        result.append(cur)
        # print(result)
        np.savetxt('output/batchsize_speed.csv', np.array(result), delimiter = ',')

readline()方法

迭代读；每读一行，指针就会向下移动。