Python:命令行,sys.argv,“如果__name__ =='__main__'”

时间:2020-10-01 07:14:44

标签: python-3.x command-line argv

我在Jupyter中使用Python有一定的经验,但是对如何使用命令行一无所知。我有完成作业的提示-我了解算法的工作原理,但是我不知道如何格式化所有内容,以便它可以按指定的方式从命令行运行。

提示:

问题1:80分

输入:指定旅行问题的文本文件(请参阅travel-input.txt) 格式)和搜索算法 (更多详细信息在下面)。

np.vectorize应该阅读 从“文件”中解决旅行问题,然后运行“搜索”算法以查找 一个办法。它将打印解决方案及其成本。

搜索是其中一项 [DFTS,DFGS,BFTS,BFGS,UCTS,UCGS,GBFTS,GBFGS,ASTS,ASGS]

这是给我的模板:

A = np.array([
[[1, 2, 3], [4, 5, 6]],
[[1, 2, 1], [3, 2, 4]]
])

N = np.array([
[2, 1],
[3, 2]
])

# desired result using a for loop:
desired_result = np.full(N.shape, np.nan)
for a in range(A.shape[0]):
    for b in range(A.shape[1]):
        # multiply the first n=N[a, b] elements
        desired_result[a, b] = np.product(A[a, b][:N[a, b]])

print(desired_result)
# output = array([[2., 4.], [2., 6.]])

python map.py [file] [search]而言,“文件”指的是from search import ... # TODO import the necessary classes and methods import sys if __name__ == '__main__': input_file = sys.argv[1] search_algo_str = sys.argv[2] # TODO implement goal_node = ... # TODO call the appropriate search function with appropriate parameters # Do not change the code below. if goal_node is not None: print("Solution path", goal_node.solution()) print("Solution cost", goal_node.path_cost) else: print("No solution was found.") ,“搜索”指的是python map.py [file] [search]之一...等等-用户指定的选择。我的问题:

  1. 我在哪里放置搜索功能?它们是否应该都在同一代码块中背靠背?
  2. 如何获取命令行以从其四个或五个字母的代码中识别每个功能?它只是函数的名称吗?如果仅使用这些字母来调用它,函数将如何接收输入?
  3. 我需要在代码中的任何地方引用输入文件吗?
  4. 我将文件保存在哪里以便从命令行访问它们-.py文件,travel-input.txt等是否重要?我尝试从命令行访问它们,但没有成功。

感谢您的帮助!

1 个答案:

答案 0 :(得分:1)

函数定义在if __name__ == "__main__"块之前。要选择正确的功能,您可以将它们放在字典中,并使用四个字母的缩写作为键,即

def dfts_search(...):
    ...

def dfgs_search(...):
    ....

...

if __name__ == "__main__":

    input_file = sys.argv[1]
    search_algo_str = sys.argv[2]
    
    search_dict = {"DFTS": dfts_search, "DFGS": dfgs_search, ...}

    try:
        func = search_dict[search_algo_str]
        result = func(...)
    except KeyError:
        print(f'{search_algo_str} is an unknown search algorithm')

不确定引用的含义,但是input_file已经引用了输入文件。您将需要编写一个函数来读取文件并处理内容。

文件的位置应该没有太大关系。将所有内容放在同一目录中可能是最简单的。在命令窗口中,仅cd到文件所在的目录,然后按照作业中的说明运行脚本。