python在txt文件的每一行增加关键字

 

import os

def updateFile(file,str):
    """
    在文件的每一行前面增加关键字
    :param file:文件名
    :param str:关键字
    """
    file_data = ""
    with open(file, "r") as f:
        for line in f:
            line = str + line 
            file_data += line
    with open(file,"w") as f:
        f.write(file_data)

base_dir = "./labels"
str  = "head,"
for file_name in os.listdir(base_dir):
    file_name = os.path.join(base_dir, file_name)
    print(file_name)
    updateFile(file_name, str)