python3 条件演算子について

Question

taisyou = [ 1 if i =="!" else 10 if i == "?" else 0 if i == "+" else i for i in (list(input()))]
print(taisyou) #[1, 10, 0, 1, 1, 10, 10, 0, 1, 1, 10, 10]
print(sum(taisyou)) #55

#入力データ !?+!!??+!!??

『概要』
入力データの内、
・! のときは、リストに1
・? のときは、リストに10
・+ のときは、リストに0
最後にsumを使って合計を出しています。

『質問』
入力データの内必要なのは、!と?だけです。
+については不要なので、無視したいのですが、やり方がわからず、ifをつかって仕方なく0を入れるようにしています。
今回に関して、不要なのは、+だけだったので、ifを書いて解決しましたが、今後不要な入力データが増えたとき、その数だけifを書くのは手間と思いました。
下記のようなことをやりたいのですが、いい方法はありますでしょうか？

taisyou = [ 1 if i =="!" else 10 if i == "?" else i for i in (list(input()))] 
#!と?だけifでかく。ほかの入力データは不要。こんなコードにしたいが、これだと望んだ動きをしない。

print(taisyou) #[1, 10, 1, 1, 10, 10, 1, 1, 10, 10] #希望する出力
print(sum(taisyou)) #55

#入力データ !?+!!??+!!??

Comments

[yohhoy]

リスト内包表記([～])だけで完結させたいということでしょうか？あまりに複雑な条件の場合、素直にメソッドを作ったほうが良いと思いますが、制約がありますか？

Answer 1

変換元の文字数が1文字限定ならstr.maketransが使えるかと。

s = input()
table = str.maketrans({
    '!': '1 ',
    '?': '10 ',
    '+': ' ',
})
result = s.translate(table)
ar = list(map(int, result.strip().split()))
print(ar)
print(sum(ar))

◇参考情報
str.maketrans

Answer 2

[ 1 if i =="!" else 10 if i == "?" else i for i in (list(input())) if i == "!" or i == "?"]

Comments

[zeitaku_fire]

ありがとうございます。末尾に対象を記載（今回では、!と?)することで、ほかは無視されるコードと解釈しました。今後ともよろしくお願いします。

Answer 3

細かいところですが、(list(input()))の外側のlistとかっこ不要かな・・・単にinput()でいいはず。

ということで、こんな感じでどうですか。
その他ならiにするという規定がなければ、条件演算子をもう1つ減らせます。

s = "!?+!!??+!!??"
taisyou = [1 if i =="!" else 10 if i == "?" else i 
           for i in s
           if i != "+"]
print(taisyou)
print(sum(taisyou))

でも、そもそも条件演算子で書くのは悪夢なので、

s = "!?+!!??+!!??"
conv_dict = {"!":1, "?":10}
taisyou = [conv_dict.get(i, i)  # get(i, i)でまずconv_dict[i]を試み、存在しなければiを返す 
           for i in s
           if i != "+"]
print(taisyou)
print(sum(taisyou))

さらに、除外したいものが複数ある場合、

s = "!?+!-!??-+!-!??"
conv_dict = {"!":1, "?":10}
exc_set = {"+", "-"}
taisyou = [conv_dict.get(i, i) 
           for i in s
           if i not in exc_set]
print(taisyou)
print(sum(taisyou))

Comments

[zeitaku_fire]

>>細かいところですが、(list(input()))の外側のlistとかっこ不要かな・・・単にinput()でいいはず。
ありがとうございます。inputだけだと、一行まるまる取り込まれると勘違いしておりました。。
ご指摘のとおり不要でした。削除しておきます。
また、辞書を使ってやる方法について、ご教授いただきありがとうございます。