原文鏈接

全文件讀寫

讀操作使用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) #變?yōu)閚*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不存在會自動創(chuàng)建，寫之前會清空文件
        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)

學習更多編程知識，請關注我的公眾號：

代碼的路