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()