Python对txt进行读写操作

原文链接

全文件读写

读操作使用pandas.read_csv，写操作使用data.to_csv。

import numpy
import pandas as pd

#读
data = pd.read_csv(r"/home/snowstorm/mmdetection/data/groundtruth.txt", header=None) #读取TXT:逗号分隔
#data = pd.read_csv(r"/home/snowstorm/mmdetection/data/groundtruth.txt", header=None, sep=' ') #读取TXT:空格分隔
print(data)

#写
data.to_csv("/home/snowstorm/mmdetection/data/write.txt",sep=',',header=False,index=False)

指定行列读写

import numpy

#读数据
def loadDatadet(infile,k):
    f = open(infile,'r')
    sourceInLine = f.readlines()
    dataset = []
    for line in sourceInLine:    #读数据
        temp1 = line.strip('\n')
        temp2 = temp1.split('\t')
        dataset.append(temp2)
        
    for i in range(0,len(dataset)):   #转为整数
        for j in range(k):
            dataset[i].append(int(dataset[i][j]))
        del(dataset[i][0:k])
        list = sum(dataset,[])        #变成一维数组
        posit_data = numpy.array(list).reshape(len(list)//4,4) #变为n*4的二维数组
    return posit_data


#写数据
def Writedata(negat_data):
    filename = 'C:\\Users\\DZF\\Desktop\\negative.txt'  #要写入的文件
    row = numpy.array(negat_data).shape[0]   #获取行数n
    with open(filename,'w') as f: # 若filename不存在会自动创建，写之前会清空文件
        for i in range(0,row):
            f.write(str(negat_data[i][0:12]))
            f.write("\n")
        
    
infile = 'C:\\Users\\snow\\Desktop\\positive.txt'   #要读取的文件
k = 1        #列数
posit_data = loadDatadet(infile,k)
Writedata(posit_data)

print(posit_data)

学习更多编程知识，请关注我的公众号：

代码的路