linux ３つの引数の間にカンマを入れて実行できるコードにしたい

Question

Linuxで下記のコードを作成しました。

import sys

def DecisionTree(FREQ,PAY,PRICE): 
  
  if  FREQ <= 5 and  PAY <= 5000 and FREQ <= 1:
     return 'RISK:1,FQ:0'
  if  FREQ <= 5 and  PAY <= 5000 and FREQ > 1:
     return 'RISK:0,FQ:0'
  if FREQ <= 5 and  PAY > 5000 :
     return 'RISK:1,FQ:0'
  elif PRICE <= 3000 :
     return 'RISK:0,FQ:1'
  else: 
     return 'RISK:1,FQ:1'

num1=int(sys.argv[1])
num2=int(sys.argv[2])
num3=int(sys.argv[3])

num=DecisionTree(num1,num2,num3)
print(num)

この画像のように、引数の間にカンマを入れて、上記のコードを実行するとと出力できません。
カンマが無ければ、出力ができるのですが、カンマを入れて出力できるようなコードに修正したいのですが、ご存じの方、ご回答いただけないでしょうか。

Answer 1

num1=int(sys.argv[1])
num2=int(sys.argv[2])
num3=int(sys.argv[3])

num=DecisionTree(num1,num2,num3)
print(num)

の部分を以下に変更しましょう。

args = sys.argv[1].split(",")

num1=int(args[0])
num2=int(args[1])
num3=int(args[2])

num=DecisionTree(num1,num2,num3)
print(num)

Comments

[Shin_go]

ありがとうございます。うまくいきました！
一点よろしければ教えていただけますか。

args = sys.argv[1].split(",")

これのリスト内の数値は、なぜ、１なのでしょうか？
０や２でもいいのですか？

[ppaul]

$ python DecisionTree.py AAA BBB
のように書くと
sys.argv[1]に"DecisionTree.py"、sys.argv[1]に"AAA"、sys.argv[2]に"BBB"
が入ります。
今回は、
$ python DecisionTree.py 6,1000,4000
なので、sys.argv[1]に"6,1000,4000"が入っています。
また、sys.argv[2]は存在しないのでアクセスするとエラーになります。
従って0や2ではだめです。

Answer 2

以下のようにして分割してあげるのがいいかもです

num1, num2, num3 = (int(x) for x in sys.argv[1].split(','))

Answer 3

これだと、argv[1]が"6,1000,4000"とかになるので、
x = sys.argv[1].split(",")するとカンマで3つに分割できます。

Comments

[Shin_go]

ありがとうございます。ただ、やはりエラーがでてうまく処理できないですね。。。

TypeError: DecisionTree() missing 2 required positional arguments: 'PAY' and 'PRICE'

というエラーがでてしまいました。

[otn]

？？
DecisionTreeを引数1つで呼んだみたいですけど、
ちゃんと、
num=DecisionTree(num1,num2,num3)
と書いてない？何故？？