web-dev-qa-db-ja.com

Pythonリストの引数パーサーリストまたはタプルのタプル

引数パーサーを使用して3D座標を解析しようとしているので、

--cord 1,2,3 2,4,6 3,6,9

取得します

((1,2,3),(2,4,6),(3,6,9))

私の試みは

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')]}

カンマの置き換えは何でしょうか?

36
joedborg

独自の type を追加できます。これにより、次のような追加の検証も可能になります。

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)
61
georg