ฉันจะเปิดหลายไฟล์โดยใช้“ with open” ใน Python ได้อย่างไร

ฉันต้องการเปลี่ยนไฟล์สองสามไฟล์ในครั้งเดียวถ้าฉันสามารถเขียนไฟล์ทั้งหมดได้ ฉันสงสัยว่าฉันจะสามารถรวมการโทรแบบเปิดหลายสายกับwithคำสั่งได้หรือไม่:

try:
  with open('a', 'w') as a and open('b', 'w') as b:
    do_something()
except IOError as e:
  print 'Operation failed: %s' % e.strerror

หากเป็นไปไม่ได้โซลูชันที่สง่างามของปัญหานี้จะเป็นอย่างไร

ตั้งแต่ Python 2.7 (หรือ 3.1 ตามลำดับ) คุณสามารถเขียนได้

with open('a', 'w') as a, open('b', 'w') as b:
    do_something()

ใน Python เวอร์ชันก่อนหน้าบางครั้งคุณสามารถใช้ contextlib.nested()เพื่อซ้อนตัวจัดการบริบท สิ่งนี้จะไม่ทำงานอย่างที่คาดไว้สำหรับการเปิดไฟล์หลายรายการ - ดูเอกสารประกอบที่เชื่อมโยงเพื่อดูรายละเอียด

ในกรณีที่ไม่ค่อยเกิดขึ้นที่คุณต้องการเปิดจำนวนตัวแปรของไฟล์ทั้งหมดในเวลาเดียวกันคุณสามารถใช้contextlib.ExitStackโดยเริ่มจาก Python เวอร์ชั่น 3.3:

with ExitStack() as stack:
    files = [stack.enter_context(open(fname)) for fname in filenames]
    # Do something with "files"

ส่วนใหญ่เวลาที่คุณมีชุดของไฟล์ตัวแปรคุณอาจต้องการเปิดไฟล์เหล่านั้นทีละไฟล์

เพียงแทนที่andด้วย,และคุณทำเสร็จแล้ว:

try:
    with open('a', 'w') as a, open('b', 'w') as b:
        do_something()
except IOError as e:
    print 'Operation failed: %s' % e.strerror

สำหรับการเปิดไฟล์หลายไฟล์พร้อมกันหรือสำหรับเส้นทางไฟล์ที่ยาวอาจเป็นประโยชน์ในการแยกสิ่งต่าง ๆ ออกเป็นหลายบรรทัด จากPython Style Guideตามที่ @Sven Marnach แนะนำในความคิดเห็นไปยังคำตอบอื่น:

with open('/path/to/InFile.ext', 'r') as file_1, \
     open('/path/to/OutFile.ext', 'w') as file_2:
    file_2.write(file_1.read())

การซ้อนคำพูดจะทำหน้าที่เดียวกันและในความคิดของฉันตรงไปตรงมากว่าที่จะจัดการกับ

สมมติว่าคุณมี inFile.txt และต้องการเขียนลงในไฟล์ outFile สองไฟล์พร้อมกัน

with open("inFile.txt", 'r') as fr:
    with open("outFile1.txt", 'w') as fw1:
        with open("outFile2.txt", 'w') as fw2:
            for line in fr.readlines():
                fw1.writelines(line)
                fw2.writelines(line)

แก้ไข:

ฉันไม่เข้าใจสาเหตุของการลงคะแนน ฉันทดสอบโค้ดก่อนที่จะเผยแพร่คำตอบของฉันและทำงานได้ตามต้องการ: มันเขียนลงในไฟล์ทั้งหมดของไฟล์เช่นเดียวกับคำถามที่ถาม ไม่มีการเขียนซ้ำหรือล้มเหลวในการเขียน ดังนั้นฉันอยากรู้จริง ๆ ว่าทำไมคำตอบของฉันถูกถือว่าผิดหรือไม่ดีหรืออะไรทำนองนั้น

ตั้งแต่ Python 3.3 คุณสามารถใช้ในชั้นเรียนExitStackจากcontextlibโมดูลได้อย่างปลอดภัยเปิดจำนวนข้อของไฟล์

มันสามารถจัดการแบบไดนามิกจำนวนของวัตถุตามบริบทซึ่งหมายความว่ามันจะเป็นประโยชน์โดยเฉพาะอย่างยิ่งถ้าคุณไม่ทราบว่าหลายไฟล์ที่คุณจะไปจับ

ในความเป็นจริงกรณีการใช้งานแบบบัญญัติซึ่งถูกกล่าวถึงในเอกสารประกอบคือการจัดการจำนวนไฟล์แบบไดนามิก

with ExitStack() as stack:
    files = [stack.enter_context(open(fname)) for fname in filenames]
    # All opened files will automatically be closed at the end of
    # the with statement, even if attempts to open files later
    # in the list raise an exception

หากคุณสนใจรายละเอียดต่อไปนี้เป็นตัวอย่างทั่วไปเพื่ออธิบายวิธีExitStackการทำงาน:

from contextlib import ExitStack

class X:
    num = 1

    def __init__(self):
        self.num = X.num
        X.num += 1

    def __repr__(self):
        cls = type(self)
        return '{cls.__name__}{self.num}'.format(cls=cls, self=self)

    def __enter__(self):
        print('enter {!r}'.format(self))
        return self.num

    def __exit__(self, exc_type, exc_value, traceback):
        print('exit {!r}'.format(self))
        return True

xs = [X() for _ in range(3)]

with ExitStack() as stack:
    print(len(stack._exit_callbacks)) # number of callbacks called on exit
    nums = [stack.enter_context(x) for x in xs]
    print(len(stack._exit_callbacks))

print(len(stack._exit_callbacks))
print(nums)

เอาท์พุท:

0
enter X1
enter X2
enter X3
3
exit X3
exit X2
exit X1
0
[1, 2, 3]

ด้วย python 2.6 มันใช้งานไม่ได้เราต้องใช้วิธีด้านล่างเพื่อเปิดหลายไฟล์:

with open('a', 'w') as a:
    with open('b', 'w') as b:

ตอบช้า (8 ปี) แต่สำหรับคนที่ต้องการเข้าร่วมหลายไฟล์เป็นหนึ่งเดียวฟังก์ชั่นต่อไปนี้อาจช่วยได้

def multi_open(_list):
    out=""
    for x in _list:
        try:
            with open(x) as f:
                out+=f.read()
        except:
            pass
            # print(f"Cannot open file {x}")
    return(out)

fl = ["C:/bdlog.txt", "C:/Jts/tws.vmoptions", "C:/not.exist"]
print(multi_open(fl))

2018-10-23 19:18:11.361 PROFILE  [Stop Drivers] [1ms]
2018-10-23 19:18:11.361 PROFILE  [Parental uninit] [0ms]
...
# This file contains VM parameters for Trader Workstation.
# Each parameter should be defined in a separate line and the
...