Python中的命令行输入问题

时间:2015-04-17 05:22:59

标签: python command-line

我在python中使用这个Tarjan算法来查找有向图中的电路。我尝试了所有类型的输入,但它没有用。

以下是complete code,以下是输入的代码。

if len(sys.argv) != 2:
    print "usage: echo \"v1 v2\nv1 v3\n...\" | %s num_vertices"%(sys.argv[0])

A = [[] for a in range(int(sys.argv[1]))]

for edge in sys.stdin.readlines():
    v1,v2 = edge.split(' ', 1)
    A[int(v1)].append(int(v2));

我已经提供了以下输入:

$ echo "3" "0 1 0 2 1 0 2 0 2 1" | python cycles.py
$ echo "3 0 1 0 2 1 0 2 0 2 1" | python cycles.py
$ echo "0 1\n0 2\n1 0\n1 3\n2 0\n3 0\n3 1\n3 2" | python cycles.py //as instructed.

但他们都没有工作。有人可以帮助搞清楚输入格式还是程序中有一些错误?

1 个答案:

答案 0 :(得分:1)

sys.argv的长度

sys.argv的长度始终包含脚本本身的名称。所以对于你的剧本:

>> print len(sys.argv)
3
>> print sys.argv
['cycles.py', '3', '0 1 0 2 1 0 2 0 2 1']

脚本错误

您检查sys.argv的长度是否不能超过2.当您传入2个参数时,sys.argv的长度为3。

if len(sys.argv) != 2:
    print "usage: echo \"v1 v2\nv1 v3\n...\" | %s num_vertices"%(sys.argv[0])

应该是

if len(sys.argv) != 3:
    print "usage: echo \"v1 v2\nv1 v3\n...\" | %s num_vertices"%(sys.argv[0])

使用参数调用python脚本

调用脚本的一种更易读的方法是使用以下语法:

python -i cycles.py "3" "0 1 0 2 1 0 2 0 2 1"