Python：命令行，sys.argv，“如果__name__ =='__main__'”

编程问答 2022-06-11

问题描述

我在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,1],[3,4]]
])

N = np.array([
[2,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等是否重要？我尝试从命令行访问它们，但没有成功。

感谢您的帮助！

解决方法

函数定义在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到文件所在的目录，然后按照作业中的说明运行脚本。

argvcommand-linepython-3.x