LeetCode 刷题必备：掌握 Python 在 Unix 中的重定向技巧

在 LeetCode 刷题的过程中，我们往往需要输入和输出大量的数据，而手动输入和输出是非常耗费时间的。因此，掌握 Python 在 Unix 中的重定向技巧是非常重要的。

Unix 系统下，标准输入、标准输出和标准错误输出都被视为文件。我们可以使用重定向符号来将数据流重定向到文件中，从而实现自动输入和输出。

以下是一些常用的重定向符号：

• <：将文件作为标准输入；
• >：将标准输出写入文件；
• >>：将标准输出追加到文件末尾；
• 2>：将标准错误输出写入文件；
• 2>>：将标准错误输出追加到文件末尾；
• &>：将标准输出和标准错误输出都写入文件；
• &>>：将标准输出和标准错误输出都追加到文件末尾。

下面，我们通过一个例子来演示如何使用 Python 在 Unix 中进行重定向。

假设我们需要将一个文本文件中的每一行都输出到控制台上，并将输出结果保存到另一个文件中。我们可以使用以下命令：

python3 main.py < input.txt > output.txt

其中，main.py 是我们的 Python 程序，input.txt 是输入文件，output.txt 是输出文件。这条命令将输入文件作为标准输入，将标准输出写入输出文件。

接下来，我们来看一下 main.py 的代码：

import sys

for line in sys.stdin:
    print(line.strip())

这段代码使用了 Python 的标准输入函数 sys.stdin，遍历了标准输入的每一行，并使用 print 函数将每一行输出到标准输出中。strip() 函数用于去除每一行末尾的空白字符。

现在，我们可以使用以下命令来运行程序，并查看输出结果：

python3 main.py < input.txt > output.txt
cat output.txt

其中，cat 命令用于查看文件内容。我们可以看到，输出文件中包含了输入文件的每一行。

除了使用文件作为重定向的输入和输出，我们还可以使用命令行参数作为输入和输出。以下是一个例子：

假设我们需要统计一个文本文件中每个单词出现的次数，并将结果输出到控制台上。我们可以使用以下命令：

cat input.txt | python3 main.py > output.txt

其中，cat 命令用于将输入文件的内容输出到标准输出中。| 符号表示将前一个命令的输出作为后一个命令的输入。这条命令将输入文件的内容作为标准输入，将标准输出写入输出文件。

接下来，我们来看一下 main.py 的代码：

import sys

count = {}

for line in sys.stdin:
    words = line.strip().split()
    for word in words:
        count[word] = count.get(word, 0) + 1

for word, cnt in count.items():
    print(word, cnt)

这段代码使用了 Python 的字典数据结构，统计了每个单词出现的次数，并使用 print 函数将结果输出到标准输出中。

现在，我们可以使用以下命令来运行程序，并查看输出结果：

cat input.txt | python3 main.py > output.txt
cat output.txt

我们可以看到，输出文件中包含了每个单词出现的次数。

总的来说，掌握 Python 在 Unix 中的重定向技巧是非常重要的。它可以帮助我们更加高效地进行 LeetCode 刷题。希望这篇文章能够对大家有所帮助。