web-dev-qa-db-ja.com

テキストファイルを変更する方法

Pythonを使用していますが、ファイルを削除またはコピーせずにテキストファイルに文字列を挿入したいと思います。どうやってやるの？

pythonfiletext
157
Oscar

残念ながら、ファイルを書き換えずにファイルの途中に挿入する方法はありません。前のポスターで示したように、seekを使用してファイルに追加したり、ファイルの一部を上書きしたりできますが、先頭または中央に追加する場合は、書き換える必要があります。

これはオペレーティングシステムの問題であり、Pythonの問題ではありません。すべての言語で同じです。

私が通常行うことは、ファイルから読み取り、変更を加え、myfile.txt.tmpまたはそのようなものと呼ばれる新しいファイルに書き込むことです。ファイルが大きすぎるため、ファイル全体をメモリに読み込むよりも優れています。一時ファイルが完成したら、元のファイルと同じ名前に変更します。

これは、ファイル書き込みが何らかの理由でクラッシュまたは中止した場合でも、元のファイルがそのまま残っているため、これを行うための適切で安全な方法です。

128
Adam Pierce

何をしたいかによって異なります。追加するには、「a」で開くことができます。

 with open("foo.txt", "a") as f:
     f.write("new line\n")

何かを前置したい場合は、最初にファイルから読み取る必要があります。

with open("foo.txt", "r+") as f:
     old = f.read() # read everything in the file
     f.seek(0) # rewind
     f.write("new line\n" + old) # write the new line before
94
Armin Ronacher

Inplace =を使用する場合、Python標準ライブラリの fileinput モジュールはインプレースでファイルを書き換えます1つのパラメーター：

import sys
import fileinput

# replace all occurrences of 'sit' with 'SIT' and insert a line after the 5th
for i, line in enumerate(fileinput.input('lorem_ipsum.txt', inplace=1)):
    sys.stdout.write(line.replace('sit', 'SIT'))  # replace 'sit' and write
    if i == 4: sys.stdout.write('\n')  # write a blank line after the 5th line
65
Dave

ファイルを適切に書き換えるには、多くの場合、古いコピーを変更された名前で保存します。 Unixの人々は~を追加して古いものをマークします。 Windowsユーザーは、.bakや.oldを追加したり、ファイルの名前を完全に変更したり、名前の先頭に〜を付けたりして、あらゆる種類のことを行います。

import shutil
shutil.move( afile, afile+"~" )

destination= open( aFile, "w" )
source= open( aFile+"~", "r" )
for line in source:
    destination.write( line )
    if <some condition>:
        destination.write( >some additional line> + "\n" )
source.close()
destination.close()

shutilの代わりに、次を使用できます。

import os
os.rename( aFile, aFile+"~" )
31
S.Lott

Pythonのmmapモジュールを使用すると、ファイルに挿入できます。次のサンプルは、Unixでの実行方法を示しています（Windows mmapは異なる場合があります）。これはすべてのエラー条件を処理するわけではないことに注意してください。元のファイルが破損または失われる可能性があります。また、これはユニコード文字列を処理しません。

import os
from mmap import mmap

def insert(filename, str, pos):
    if len(str) < 1:
        # nothing to insert
        return

    f = open(filename, 'r+')
    m = mmap(f.fileno(), os.path.getsize(filename))
    origSize = m.size()

    # or this could be an error
    if pos > origSize:
        pos = origSize
    Elif pos < 0:
        pos = 0

    m.resize(origSize + len(str))
    m[pos+len(str):] = m[pos:origSize]
    m[pos:pos+len(str)] = str
    m.close()
    f.close()

ファイルを 'r +'モードで開いた状態でmmapなしでこれを行うこともできますが、挿入位置からEOF-これは巨大かもしれません。

14
mhawke

Adamが述べたように、すべてをメモリに読み込むのに十分なメモリがあるかどうかを判断する前に、システムの制限を考慮に入れて、メモリの一部を置き換えて書き換える必要があります。

あなたが小さなファイルを扱っているか、メモリの問題がない場合、これは役立つかもしれません：

オプション1）ファイル全体をメモリに読み込み、行全体または行の一部で正規表現の置換を行い、その行と追加行で置き換えます。 「中間行」がファイル内で一意であることを確認する必要があります。各行にタイムスタンプがある場合、これはかなり信頼できるはずです。

# open file with r+b (allow write and binary mode)
f = open("file.log", 'r+b')   
# read entire content of file into memory
f_content = f.read()
# basically match middle line and replace it with itself and the extra line
f_content = re.sub(r'(middle line)', r'\1\nnew line', f_content)
# return pointer to top of file so we can re-write the content with replaced string
f.seek(0)
# clear file content 
f.truncate()
# re-write the content with the updated content
f.write(f_content)
# close file
f.close()

オプション2）中央の行を計算し、その行に余分な行を加えて置き換えます。

# open file with r+b (allow write and binary mode)
f = open("file.log" , 'r+b')   
# get array of lines
f_content = f.readlines()
# get middle line
middle_line = len(f_content)/2
# overwrite middle line
f_content[middle_line] += "\nnew line"
# return pointer to top of file so we can re-write the content with replaced string
f.seek(0)
# clear file content 
f.truncate()
# re-write the content with the updated content
f.write(''.join(f_content))
# close file
f.close()
12
Maxime R.

UNIXを知っているなら、以下を試すことができます：

注：$はコマンドプロンプトを意味します

次のような内容のファイルmy_data.txtがあるとします。

$ cat my_data.txt
This is a data file
with all of my data in it.

その後、osモジュールを使用すると、通常のsedコマンドを使用できます

import os

# Identifiers used are:
my_data_file = "my_data.txt"
command = "sed -i 's/all/none/' my_data.txt"

# Execute the command
os.system(command)

Sedに気付いていない場合は、チェックしてみてください。非常に便利です。

1
G. LC

これをきれいに行うための小さなクラスを書きました。

import tempfile

class FileModifierError(Exception):
    pass

class FileModifier(object):

    def __init__(self, fname):
        self.__write_dict = {}
        self.__filename = fname
        self.__tempfile = tempfile.TemporaryFile()
        with open(fname, 'rb') as fp:
            for line in fp:
                self.__tempfile.write(line)
        self.__tempfile.seek(0)

    def write(self, s, line_number = 'END'):
        if line_number != 'END' and not isinstance(line_number, (int, float)):
            raise FileModifierError("Line number %s is not a valid number" % line_number)
        try:
            self.__write_dict[line_number].append(s)
        except KeyError:
            self.__write_dict[line_number] = [s]

    def writeline(self, s, line_number = 'END'):
        self.write('%s\n' % s, line_number)

    def writelines(self, s, line_number = 'END'):
        for ln in s:
            self.writeline(s, line_number)

    def __popline(self, index, fp):
        try:
            ilines = self.__write_dict.pop(index)
            for line in ilines:
                fp.write(line)
        except KeyError:
            pass

    def close(self):
        self.__exit__(None, None, None)

    def __enter__(self):
        return self

    def __exit__(self, type, value, traceback):
        with open(self.__filename,'w') as fp:
            for index, line in enumerate(self.__tempfile.readlines()):
                self.__popline(index, fp)
                fp.write(line)
            for index in sorted(self.__write_dict):
                for line in self.__write_dict[index]:
                    fp.write(line)
        self.__tempfile.close()

その後、次のように使用できます。

with FileModifier(filename) as fp:
    fp.writeline("String 1", 0)
    fp.writeline("String 2", 20)
    fp.writeline("String 3")  # To write at the end of the file
1
ananth krishnan