根据命令行参数的数量创建多个集合

时间:2015-01-31 04:06:05

标签: python

我的程序将比较任意数量的集合。 如果金额由命令行参数的数量决定,我如何为每个参数创建一个唯一的集合?

1 个答案:

答案 0 :(得分:1)

import ast
import sys

def main(sets):
    # now do something with them

if __name__ == "__main__":
    sets = [
        set(ast.literal_eval(arg))
        for arg in sys.argv[1:]
    ]
    main(sets)

你可以像

一样打电话
python myprog.py "1, 2" "1, 5" "6, 7, 8" "12, 13"

并导致调用

main([{1, 2}, {1, 5}, {8, 6, 7}, {12, 13}])