Python Argparse: Vấn đề với các đối số tùy chọn là số âm

Question

Tôi gặp sự cố nhỏ với argparse. Tôi có một tùy chọn xlim là xrange của một âm mưu. Tôi muốn có thể chuyển số như -2e-5. Tuy nhiên, điều này không có tác dụng - argparse diễn giải đây là một đối số vị trí. Nếu tôi làm -0.00002 hoạt động: argparse sẽ đọc số này làm số âm. Có thể đọc được trong -2e-3 không?

Mã này là dưới đây, và một ví dụ về làm thế nào tôi sẽ chạy nó là:

./blaa.py --xlim -2.e-3 1e4 

Nếu tôi làm như sau hoạt động:

./blaa.py --xlim -0.002 1e4 

Mã:

parser.add_argument('--xlim', nargs = 2, 
        help = 'X axis limits', 
        action = 'store', type = float, 
        default = [-1.e-3, 1.e-3]) 

Trong khi tôi có thể làm cho nó hoạt động theo cách này, tôi thực sự sẽ có thể sử dụng ký pháp khoa học. Ai có ý tưởng gì không?

Cheers

Answer 1

Như đã chỉ ra bởi các ý kiến, vấn đề là một tiền tố - được phân tách như một tùy chọn thay vì là một cuộc tranh cãi. One way để workaround this is change các prefix used for options with prefix_chars argument:

#!/usr/bin/python 
import argparse 

parser = argparse.ArgumentParser(prefix_chars='@') 
parser.add_argument('@@xlim', nargs = 2, 
        help = 'X axis limits', 
        action = 'store', type = float, 
        default = [-1.e-3, 1.e-3]) 
print parser.parse_args() 

Example output:

$ ./blaa.py @@xlim -2.e-3 1e4 
Namespace(xlim=[-0.002, 10000.0]) 

Edit: Alternatively, bạn can keep using - as separator, pass xlim as a single giá trị và sử dụng hàm trong type để triển khai phân tích cú pháp của riêng bạn:

#!/usr/bin/python 
import argparse 

def two_floats(value): 
    values = value.split() 
    if len(values) != 2: 
     raise argparse.ArgumentError 
    values = map(float, values) 
    return values 

parser = argparse.ArgumentParser() 
parser.add_argument('--xlim', 
        help = 'X axis limits', 
        action = 'store', type=two_floats, 
        default = [-1.e-3, 1.e-3]) 
print parser.parse_args() 

Ví dụ outp ut:

$ ./blaa.py --xlim "-2e-3 1e4" 
Namespace(xlim=[-0.002, 10000.0]) 

Answer 2

Nếu bạn đang lên đến sửa đổi argparse.py bản thân, bạn có thể thay đổi khớp số âm để xử lý ký hiệu khoa học:

Trong class _ActionsContainer.__init__()

self._negative_number_matcher = _re.compile(r'^-(\d+\.?|\d*\.\d+)([eE][+\-]?\d+)?$') 

Hoặc sau khi tạo phân tích cú pháp, bạn có thể đặt parser._negative_number_matcher thành giá trị này. Cách tiếp cận này có thể có vấn đề nếu bạn đang tạo nhóm hoặc subparsers, nhưng nên làm việc với một trình phân tích cú pháp đơn giản.

Answer 3

Một giải pháp thay thế mà tôi đã tìm thấy là báo giá trị, nhưng thêm khoảng trắng. Đó là,

./blaa.py --xlim " -2.e-3" 1e4 

cách argparse này sẽ không nghĩ -2.e-3 là một tên tùy chọn vì ký tự đầu tiên không phải là một gạch nối-dash, nhưng nó vẫn sẽ được chuyển đổi đúng cách để một phao vì phao (chuỗi) bỏ qua dấu cách ở bên trái.

Answer 4

Đây là mã tôi sử dụng. (Nó tương tự như jeremiahbuddha nhưng nó trả lời câu hỏi trực tiếp hơn vì nó đề cập đến các số âm.)

Đặt này trước khi gọi argparse.ArgumentParser()

for i, arg in enumerate(sys.argv): 
    if (arg[0] == '-') and arg[1].isdigit(): sys.argv[i] = ' ' + arg 

Answer 5

workaround khác là để vượt qua trong đối số sử dụng '=' biểu tượng ngoài việc trích dẫn các đối số - tức là, --xlim="-2.3e14"

Answer 6

Lấy cảm hứng từ cách tiếp cận andrewfn, tôi đã tạo một hàm trợ giúp riêng biệt để thực hiện chức năng số sys.argv:

def _tweak_neg_scinot(): 
    import re 
    import sys 
    p = re.compile('-\\d*\\.?\\d*e', re.I) 
    sys.argv = [' ' + a if p.match(a) else a for a in sys.argv] 

Các regex tìm kiếm:

• -: một dấu hiệu tiêu cực
• \\d*: zero hoặc nhiều chữ số (cho các giá trị kỳ quặc định dạng giống như -.5e-2 hoặc -4354.5e-6)
• \\.?: một khoảng thời gian tùy chọn (ví dụ, -2e-5 là hợp lý)
• \\d*: một bộ số không hoặc nhiều chữ số khác (cho những thứ như -2e-5 và -7.e-3)
• e: để khớp với điểm đánh dấu số mũ

re.I làm cho nó khớp với cả hai -2e-5 và -2E-5. Sử dụng p.match có nghĩa là nó chỉ tìm kiếm từ đầu mỗi chuỗi.

Answer 7

Nếu bạn chỉ định giá trị cho lựa chọn của bạn với một dấu bằng, argparse sẽ không đối xử với nó như một lựa chọn riêng biệt, thậm chí nếu nó bắt đầu với -:

./blaa.py --xlim='-0.002 1e4' 
# As opposed to --xlim '-0.002 1e4' 

Và nếu giá trị không có khoảng trống trong nó, bạn có thể thả các dấu ngoặc kép:

./blaa.py --xlim=-0.002 

Xem: https://www.gnu.org/software/guile/manual/html_node/Command-Line-Format.html

với điều này, không có cần phải viết bạn r own type= trình phân tích cú pháp hoặc xác định lại ký tự tiền tố từ - đến @ làm câu trả lời được chấp nhận.