Question

所以我有这些函数用于将输出打印/写入文本文件。现在，当这些函数在python shell中打印输出时，我得到了我需要的东西。当我尝试将这些输出写入文本文件时，事情就会出错。

以下是功能：

def printsection1(animals, station1, station2):
    for animal in animals:
        print(animal,'                 ',station1.get(animal, 0),'                   ',station2.get(animal, 0))

def printsection2(animals, station1, station2):
    for animal in animals:
        if station2.get(animal, 0)+station1.get(animal, 0) >= 4:
            print(animal)

def printsection3(animals, station1, station2):
    for animal in animals:
        print(animal,'                    ',int(station1.get(animal, 0))+int(station2.get(animal, 0)))

def printsection4(items):
    import statistics
    most_visits=[]
    for animal, date, station in items:
        most_visits.append(date[0:2]) 

    print(statistics.mode(most_visits))

我在main（）函数中写入文本文件的代码如下所示：

outfile.write("Number of times each animal visited each station:\n")
outfile.write("Animal ID           Station 1           Station 2 \n")
printsection1(animals, station1, station2)
outfile.write('\n')
outfile.write("============================================================ \n")

outfile.write("Animals that visited both stations at least 4 times \n")
printsection2(animals, station1, station2)
outfile.write('\n')
outfile.write("============================================================ \n")

outfile.write("Total number of visits for each animal \n")
printsection3(animals, station1, station2)
outfile.write('\n')
outfile.write("============================================================ \n")

outfile.write("Month that has the highest number of visits to the stations \n")
printsection4(items)

是否有一种简单的方法可以将函数的输出写入文本文件？我看过“＆gt;＆gt;”操作员漂浮，但我似乎无法让它工作。如果需要更多信息，我可以提供。

再次感谢！

Answer 1

您还需要将函数中的print调用更改为outfile.write，或使用print(..., file=outfile)语法。

或者，您可以使用作业outfile将标准输出点设为sys.stdout = outfile。这将导致对print的所有后续调用都写入outfile。

Answer 2

在python 3中，您可以将print函数的输出重定向到包含file关键字的文件：

with open('somefile.txt', 'rt') as f:
  print('Hello World!', file=f)

将python函数输出写入.txt文件

2 个答案: