python在txt文件中添加内容_如何在Python的.txt文件中添加头

检查标题行是否已存在，如果第一行中不存在，则将其写入文件。name = input ("Name: ")

age = input("Age: ")

BirthYear = input("Birth Year: ")

filename = "info.txt"

header = "Name Age Grade\n"

def WriteHeader(filename, header):

"""

;param filename: a file path

;param header: a string representing the file's "header" row

This function will check if the header exists in the first line

and inserts the header if it doesn't exist

"""

file = open(filename, 'r')

lines = [line for line in file]

file.close()

if lines and lines[0] == header:

# There are some lines in the file, and first line is the header

return True

else:

# The first line is NOT the header

file = open(filename, w)

# Rewrite the file: append header if need