python 文件插入第一行

最新推荐文章于 2024-01-06 20:48:12 发布

原创最新推荐文章于 2024-01-06 20:48:12 发布 · 1.9k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

Python 专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一个Python函数，用于将多个字符串插入到文件中指定的行号处，实现高效文本编辑。通过设置行号和要插入的字符串，用户可以轻松修改大量文本文件的内容。

def file_insert(fname,linenos=[],strings=[]):  
    """ 
    Insert several strings to lines with linenos repectively. 
 
    The elements in linenos must be in increasing order and len(strings) 
    must be equal to or less than len(linenos). 
 
    The extra lines ( if len(linenos)> len(strings)) will be inserted 
    with blank line. 
    """  
    if os.path.exists(fname):  
        lineno = 0  
        i = 0  
        for line in fileinput.input(fname,inplace=1):  
            # inplace must be set to 1  
            # it will redirect stdout to the input file  
            lineno += 1  
            line = line.strip()  
            if i<len(linenos) and linenos[i]==lineno:  
                if i>=len(strings):  
                    print "\n",line  
                else:  
                    print strings[i]  
                    print line  
                i += 1  
            else:  
                print line

fileinput.input的inplace必须要设为1，以便让stdout被重定向到输入文件里：

file_insert('a.txt' ,[ 1 , 4 , 5 ],[ 'insert1' , 'insert4' ]