Python 如何从命令行参数获取文件名

Question

我有以下代码：

with open("test.txt", "r") as test, open("table.txt", "w") as table:
    reader = csv.reader(test, delimiter="\t")
    writer = csv.writer(table, delimiter="\t")
    for row in reader:
        if all(field not in keywords for field in row):
            writer.writerow(row)

我如何才能将它转换为 .py 文件，让您定义 table.txt 并在运行时进行测试。所以必须写：

Script.py test.txt output.txt

Answer 1

只需使用sys.argv:

import sys
import csv

with open(sys.argv[1], "r") as test, open(sys.argv[2], "w") as table:
      # more here

注意，sys.argv[0] 包含脚本名称（在您的例子中，Script.py）。要得到第一个参数，你应该得到 sys.argv[1];要得到第二个参数，你应该得到 sys.argv[2] 等等。

Python 如何从命令行参数获取文件名

How to get file names from command-line parameters in Python

python

input

output