Python 等价运算符没有按我预期的方式工作

Question

我有一个 Python 脚本，它在从命令行运行脚本时将命令行参数分配给不同的变量。我以前做过这个，几乎没有问题，但这次我被一些似乎有点技术性的东西挂断了，我不能轻易解决。

假设我在开头附近有以下语句：

n_value = sys.argv[3]

我期望的命令行参数是一个整数字符串，从 1 到 6。然后，稍后，我想测试 n_value 指向的值，以确定下一步要走的路。所以，我有以下几点：

if n_value == "1":
    (do something)

最后，对于不期望命令行参数输入的实例，有一个 else 语句。每次我尝试运行程序时都会得到这个。我尝试将 if 语句更改为：

if n_value == 1:

或者：

if n_value is "1"

等等。我已经尝试了很多东西，但我似乎无法实现该值。

我也尝试使用 pdb，在这些语句之前使用 set_trace()。在调试器中，我尝试查看表达式 n_value == "1" 的值，它显示为“True”。这让我相信这可能是我正在使用的版本的一些技术问题（即，我做错了什么但我没有意识到）或者我只是不了解 Python 等价的来龙去脉操作。

最后一点：我一直使用的 Python 版本是 2.6 和 2.7。据我所知，两者都存在同样的问题。

如果有兴趣，请看下面我的 main() 方法的开头：

def main():
    if len(sys.argv) != 4:
        print(r'usage: python(2.6) ./log_likelihood_ngrams.py /path/to/input_file1 /path/to/input_file2 n_value')
        sys.exit(1)

    # Store command-line arguments as variables
    input_file1_path = sys.argv[1]
    input_file2_path = sys.argv[2]
    n_value = sys.argv[3]

    # Tokenize the input files and save their n-grams in n-gram-lists

    # For 1-grams
    if n_value == '1':
        ngrams_list1 = tokenize(input_file1_path)
        ngrams_list2 = tokenize(input_file2_path)
    # For 2-grams
    if n_value == '2':
        ngrams_list1 = bigram_list(input_file1_path)
        ngrams_list2 = bigram_list(input_file2_path)
    # For 3-grams
    if n_value == '3':
        ngrams_list1 = trigram_list(input_file1_path)
        ngrams_list2 = trigram_list(input_file2_path)
    # For 4-grams
    if n_value == '4':
        ngrams_list1 = four_gram_list(input_file1_path)
    ngrams_list2 = four_gram_list(input_file2_path)
    # For 5-grams
    if n_value == '5':
        ngrams_list1 = five_gram_list(input_file1_path)
        ngrams_list2 = five_gram_list(input_file2_path)
    # For 6-grams
    if n_value == '6':
        ngrams_list1 = six_gram_list(input_file1_path)
        ngrams_list2 = six_gram_list(input_file2_path)
    # If n is invalid, print an error message and exit the program.
    else:
        sys.stderr.write('\n\nThe value of n you entered is not valid!\nPlease enter a value between 1 and 6, inclusive.\n')
        sys.exit(1)

Answer 1

你想要一个elif 链。你所拥有的是一系列if 语句；因此，即使if n_value == '1' 检查通过，所有其他检查仍将运行。当if n_value == '6' 失败时，附加的else 将运行，这是您不想要的。使用elif:

if n_value == '1':
    ...
elif n_value == '2':
    ...
...
else:
    whatever

Answer 2

if n_value == '1' 之后的所有if 语句都应该是elif (http://docs.python.org/2/tutorial/controlflow.html)。

例如

if n_value == '1':
    # For 1-grams
    ngrams_list1 = tokenize(input_file1_path)
    ngrams_list2 = tokenize(input_file2_path)
elif n_value == '2':
    # For 2-grams
    ngrams_list1 = bigram_list(input_file1_path)

... etc ...

else:
    sys.stderr.write('\n\nThe value of n you entered is not valid!\nPlease enter a value between 1 and 6, inclusive.\n')
    sys.exit(1)