Python łączy pliki tekstowe

Mam listę 20 nazw plików, na przykład ['file1.txt', 'file2.txt', ...]. Chcę napisać skrypt w języku Python, aby połączyć te pliki w nowy plik. Mógłbym otworzyć każdy plik przez f = open(...), przeczytać wiersz po wierszu przez wywołanie f.readline()i zapisać każdy wiersz w tym nowym pliku. Nie wydaje mi się to zbyt „eleganckie”, zwłaszcza część, w której muszę czytać // pisać wiersz po wierszu.

Czy jest bardziej „elegancki” sposób na zrobienie tego w Pythonie?

To powinno wystarczyć

W przypadku dużych plików:

filenames = ['file1.txt', 'file2.txt', ...]
with open('path/to/output/file', 'w') as outfile:
    for fname in filenames:
        with open(fname) as infile:
            for line in infile:
                outfile.write(line)

W przypadku małych plików:

filenames = ['file1.txt', 'file2.txt', ...]
with open('path/to/output/file', 'w') as outfile:
    for fname in filenames:
        with open(fname) as infile:
            outfile.write(infile.read())

… I kolejny ciekawy, o którym pomyślałem :

filenames = ['file1.txt', 'file2.txt', ...]
with open('path/to/output/file', 'w') as outfile:
    for line in itertools.chain.from_iterable(itertools.imap(open, filnames)):
        outfile.write(line)

Niestety, ta ostatnia metoda pozostawia kilka otwartych deskryptorów plików, którymi i tak powinien się zająć GC. Po prostu pomyślałem, że to interesujące

Użyj shutil.copyfileobj.

Automatycznie czyta pliki wejściowe fragment po kawałku dla Ciebie, co jest bardziej wydajne i wczytuje pliki wejściowe i będzie działać, nawet jeśli niektóre pliki wejściowe są zbyt duże, aby zmieścić się w pamięci:

import shutil

with open('output_file.txt','wb') as wfd:
    for f in ['seg1.txt','seg2.txt','seg3.txt']:
        with open(f,'rb') as fd:
            shutil.copyfileobj(fd, wfd)

Właśnie do tego służy fileinput :

import fileinput
with open(outfilename, 'w') as fout, fileinput.input(filenames) as fin:
    for line in fin:
        fout.write(line)

W tym przypadku nie jest to o wiele prostsze niż tylko ręczne iterowanie po plikach, ale w innych przypadkach posiadanie jednego iteratora, który iteruje wszystkie pliki, tak jakby były jednym plikiem, jest bardzo przydatne. (Ponadto fakt, że fileinputzamyka każdy plik zaraz po zakończeniu, oznacza, że ​​nie ma takiej potrzeby withlub closekażdego z nich, ale to tylko jedna linia oszczędności, a nie taka wielka sprawa).

W programie jest kilka innych przydatnych funkcji fileinput, takich jak możliwość wykonywania lokalnych modyfikacji plików, po prostu filtrując każdą linię.

Jak zauważono w komentarzach i omówiono w innym poście , fileinputPython 2.7 nie będzie działał zgodnie z opisem. Tutaj niewielka modyfikacja, aby kod był zgodny z Pythonem 2.7

with open('outfilename', 'w') as fout:
    fin = fileinput.input(filenames)
    for line in fin:
        fout.write(line)
    fin.close()

Nie znam elegancji, ale to działa:

    import glob
    import os
    for f in glob.glob("file*.txt"):
         os.system("cat "+f+" >> OutFile.txt")

Co jest nie tak z poleceniami UNIX? (zakładając, że nie pracujesz w systemie Windows):

ls | xargs cat | tee output.txt wykonuje zadanie (jeśli chcesz, możesz to wywołać z Pythona z podprocesem)

outfile.write(infile.read()) # time: 2.1085190773010254s
shutil.copyfileobj(fd, wfd, 1024*1024*10) # time: 0.60599684715271s

Prosty test porównawczy pokazuje, że shutil działa lepiej.

Alternatywa dla odpowiedzi @ inspectorG4dget (najlepsza odpowiedź do tej pory 29-03-2016). Testowałem z 3 plikami po 436 MB.

@ inspectorG4dget rozwiązanie: 162 sekundy

Następujące rozwiązanie: 125 sekund

from subprocess import Popen
filenames = ['file1.txt', 'file2.txt', 'file3.txt']
fbatch = open('batch.bat','w')
str ="type "
for f in filenames:
    str+= f + " "
fbatch.write(str + " > file4results.txt")
fbatch.close()
p = Popen("batch.bat", cwd=r"Drive:\Path\to\folder")
stdout, stderr = p.communicate()

Chodzi o to, aby utworzyć plik wsadowy i uruchomić go, wykorzystując „starą dobrą technologię”. Jego semi-Python, ale działa szybciej. Działa dla okien.

Jeśli masz dużo plików w katalogu, glob2lepszym rozwiązaniem może być wygenerowanie listy nazw plików, zamiast wpisywania ich ręcznie.

import glob2

filenames = glob2.glob('*.txt')  # list of all .txt files in the directory

with open('outfile.txt', 'w') as f:
    for file in filenames:
        with open(file) as infile:
            f.write(infile.read()+'\n')

Sprawdź metodę .read () obiektu File:

http://docs.python.org/2/tutorial/inputoutput.html#methods-of-file-objects

Możesz zrobić coś takiego:

concat = ""
for file in files:
    concat += open(file).read()

lub bardziej „elegancki” sposób w Pythonie:

concat = ''.join([open(f).read() for f in files])

co zgodnie z tym artykułem: http://www.skymind.com/~ocrow/python_string/ byłoby również najszybsze.

Jeśli pliki nie są gigantyczne:

with open('newfile.txt','wb') as newf:
    for filename in list_of_files:
        with open(filename,'rb') as hf:
            newf.write(hf.read())
            # newf.write('\n\n\n')   if you want to introduce
            # some blank lines between the contents of the copied files

Jeśli pliki są zbyt duże, aby można je było w całości odczytać i przechowywać w pamięci RAM, algorytm musi być nieco inny, aby odczytać każdy plik, który ma być skopiowany w pętli przez fragmenty o ustalonej długości, read(10000)na przykład.

def concatFiles():
    path = 'input/'
    files = os.listdir(path)
    for idx, infile in enumerate(files):
        print ("File #" + str(idx) + "  " + infile)
    concat = ''.join([open(path + f).read() for f in files])
    with open("output_concatFile.txt", "w") as fo:
        fo.write(path + concat)

if __name__ == "__main__":
    concatFiles()

  import os
  files=os.listdir()
  print(files)
  print('#',tuple(files))
  name=input('Enter the inclusive file name: ')
  exten=input('Enter the type(extension): ')
  filename=name+'.'+exten
  output_file=open(filename,'w+')
  for i in files:
    print(i)
    j=files.index(i)
    f_j=open(i,'r')
    print(f_j.read())
    for x in f_j:
      outfile.write(x)