Python参数解析器列表或元组元组

jdb*_*org 35 python argparse

我正在尝试使用参数解析器来解析3D坐标,以便我可以使用

--cord 1,2,3 2,4,6 3,6,9
Run Code Online (Sandbox Code Playgroud)

得到

((1,2,3),(2,4,6),(3,6,9))
Run Code Online (Sandbox Code Playgroud)

我的尝试是

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--cord', help="Coordinate", dest="cord", type=tuple, nargs=3)
args = parser.parse_args(["--cord","1,2,3","2,4,6","3,6,9"])

vars(args) 
{'cord': [('1', ',', '2', ',', '3'),
  ('2', ',', '4', ',', '6'),
  ('3', ',', '6', ',', '9')]}
Run Code Online (Sandbox Code Playgroud)

替换逗号会是什么?

geo*_*org 59

您可以添加自己的类型.这还允许进行其他验证,例如:

def coords(s):
    try:
        x, y, z = map(int, s.split(','))
        return x, y, z
    except:
        raise argparse.ArgumentTypeError("Coordinates must be x,y,z")


parser.add_argument('--cord', help="Coordinate", dest="cord", type=coords, nargs=3)
Run Code Online (Sandbox Code Playgroud)

  • 由于这种情况是“3D”坐标,“nargs=3”是显而易见的,但是如果您的情况不需要固定数量的元组,则可以使用“nargs='+'” (2认同)