Python 键错误 : 'OUTPUT_PATH'

Question

我正在尝试运行以下 python 代码来练习

#!/bin/python3

import os
import sys

#
# Complete the maximumDraws function below.
#
def maximumDraws(n):
    return n+1

if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    t = int(input())

    for t_itr in range(t):
        n = int(input())

        result = maximumDraws(n)

        fptr.write(str(result) + '\n')

    fptr.close()

但是我收到这个错误信息

Traceback (most recent call last):
  File "maximumdraws.py", line 13, in <module>
    fptr = open(os.environ['OUTPUT_PATH'], 'w')
  File "/home/inindekikral/anaconda3/lib/python3.6/os.py", line 669, in __getitem__
    raise KeyError(key) from None
KeyError: 'OUTPUT_PATH'

我的操作系统是 Linux Mint 19 Cinnamon。我必须做什么？

Answer 1

我确定还有其他方法可以做到这一点，但是对于 Hackerrank 练习，文件指针是这样打开的:

fptr = open(os.environ['OUTPUT_PATH'], 'w')

...我希望它直接进入标准输出。

我只是把那行改成了

fptr = sys.stdout   # stdout is already an open stream

它做我想做的事。

请注意，一方面，os.environ['OUTPUT_PATH'] 是一个字符串，而 fptr 是一个流/文件指针。

变化:

1. 如果您想写入文件，可以按照上面建议的方式进行(设置 OUTPUT_PATH 环境变量)。

2. 或者，您可以直接在 python 中设置 os.environ，例如

   os.environ['OUTPUT_PATH'] = 'junk.txt' # 在打开 fptr 之前!