Python（五）文件操作基礎代碼示例

操作示例代碼1，讀取文件：

def main():
    fname = input("Enter filename:")
    infile = open(fname,"r")
    data = infile.read()
    print(data)
    infile.close()
main()

操作示例代碼2，讀取文件：

def main():
    infile = open("xx/someFile.txt","r")
    for i in range(5):
        line = infile.readline()
        print(line[:-1])
    infile.close()

main()

文件寫入示例代碼：

文件操作實例（一）

讀取數據文件，使用turtle庫繪製圖形：

1，data.txt文件數據：

300，0，144，1，0，0
300，0，144，0，1，0
300，0，144，0，0，1
300，0，144，1，1，0
300，0，108，0，1，1
184，0， 72，1，0，1
184，0， 72，0，0，0
184，0， 72，0，0，0
184，0， 72，0，0，0

2，data.txt文件數據按列表元素的意義：

元素1：路徑前進像素數。
元素2：轉動方向，0爲左，1爲右。
元素3：轉動角度。
元素4-6：繪製顏色的rgb值

3，調用的Python文件：

# dothedocument4.py
# encoding=utf-8
import turtle

def main():
    turtle.title("數據驅動的動態路徑繪製")
    turtle.setup(800,600,0,0)
    #設置畫筆
    pen = turtle.Turtle()
    pen.color("red")
    pen.width(5)
    pen.shape("turtle")
    pen.speed(5)
    
    #讀取文件
    result=[]
    file = open("data.txt","r")
    for line in file:
        result.append(list(map(float,line.split('，'))))
    print(result)

    #動態繪製
    for i in range(len(result)):
        pen.color((result[i][3],result[i][4],result[i][5]))
        pen.fd(result[i][0])
        if result[i][1]:
            pen.rt(result[i][2])
        else:
            pen.lt(result[i][2])
    pen.goto(0,0)
    turtle.done()

main()

運行結果：

文件操作實例（二）：多文件讀寫：

1，數據文件和目標文件：

操作目的：文件EmailAddressBook.txt和文件TeleAddressBook.txt二者合併，得到文件AddRessBook.txt,記錄姓名、郵箱、電話，相同名字的郵箱和電話，視爲同一人。代碼如下：

# dothedocument5.py
# encoding=utf-8
# 多文件讀寫
def main():
    #打開文件、讀取文件
    ftele1=open('TeleAddressBook.txt','rb')#由於本文件出現中文，所以使用’rb’避免亂碼
    ftele2=open('EmailAddressBook.txt','rb')

    ftele1.readline()#跳過第一行
    ftele2.readline()
    lines1 = ftele1.readlines()
    lines2 = ftele2.readlines()

    # 獲取TeleAddressBook中的信息
    list1_name = []
    list1_tele = []
    list2_name = []
    list2_email = []
    
    for line in lines1:#獲取第一個文本的姓名和電話信息
        elements = line.split()
        list1_name.append(str(elements[0].decode('gbk')))
        list1_tele.append(str(elements[1].decode('gbk')))

    # 獲取EmailAddressBook中的信息
    for line in lines2:#獲取第二個文本中的姓名和信息
        elements = line.split()
        list2_name.append(str(elements[0].decode('gbk')))
        list2_email.append(str(elements[1].decode('gbk')))

    lines = []
    lines.append('姓名\t	電話\t	郵箱\n')# \t表示空格

    # 按索引方式遍歷列表1
    for i in range(len(list1_name)):
        s = ''
        if list1_name[i] in list2_name:
            j = list2_name.index(list1_name[i]) # 找到姓名列表1對應列表2中的姓名
            s = '\t'.join([list1_name[i],list1_tele[i],list2_email[j]])
            s += '\n'
        else:
            s = '\t'.join([list1_name[i],list1_tele[i],str('   ----    ')])
            s += '\n'
        lines.append(s)

    # 處理姓名列表2中剩餘的姓名
    for i in range(len(list2_name)):
        s = ''
        if list2_name not in list1_name:
            s = '\t'.join([list2_name[i],str('   ----    '),list2_email[i]])
            s += '\n'
        lines.append(s)

    # 將新生成的合併數據寫入新的文件中
    ftele3 = open('AddressBook.txt','w')
    ftele3.writelines(lines)

    # 關閉文件
    ftele3.close()
    ftele1.close()
    ftele2.close()

    print("The addressBooks are merged!")

main()