re.sub()を複数回書かず、一括で複数条件文字置換がしたい

Question

聞きたいこと

• 私のコードだとre.sub()を2回書いているので、1回または1行の処理で済ませるスマートな書き方が知りたい。（または不可能な場合は、不可能であるということを知りたい）
• re.sub()は使用しなくてもいいですが、正規表現を使いたい。

前提条件

• '1'と'0'から構成される文字列を戻す関数をつくる。
• 引数にはランダムな文字列が与えられる。
• 母音（aeiouAEIOU）は'1'、それ以外の全て（空白なども含む）は'0'に置き換える。

例

引数に与えられる値	戻り値
'amEricA'	'1010101'
'123abc'	'000100'
'Hello, world!'	'0100100010000'

該当コード

import re

def vowel_one(str):
    str = re.sub(r'[^aeiou]', '0', str, flags=re.IGNORECASE)
    str = re.sub(r'[^\d]', '1', str) # 問題の部分。2行にわたってre.subを書いていて冗長的に感じる
    return str

試したこと

正規表現を使用しない別バージョンは、1行で書けました。

def vowel_one(str):
    return ''.join('1' if c in 'aeiou' else '0' for c in str.lower())

Comments

[maisumakun]

> 正規表現を使いたい。

どういう意図があってでしょうか？

[Cocode]

コメントありがとうございます。正規表現とそのメソッドの練習のためです。

Answer 1

import re

def vowel_one(text):
    text = re.sub(r'(?i)([aeiou])|(.)', lambda m: '1' if m[1] else '0', text)
    return text

if __name__ == '__main__':
    test_data = [
      ['amEricA', '1010101'],
      ['123abc', '000100'],
      ['Hello, world!', '0100100010000'],
    ]

    for text, expected in test_data:
        print(f'{text} =>\n    output: {vowel_one(text)}\n  expected: {expected}\n')

#
amEricA =>
    output: 1010101
  expected: 1010101

123abc =>
    output: 000100
  expected: 000100

Hello, world! =>
    output: 0100100010000
  expected: 0100100010000

Comments

[Cocode]

回答ありがとうございますm(_ _)m

そして質問失礼します！
m[1]とはどういう意味ですか？

[melian]

正規表現 (?i)([aeiou]) にマッチする文字になります。マッチしない場合、m[1] は None になりますので 、

lambda m: '1' if m[1] else '0'

ここで "0" に置換されることになります。

Answer 2

import re

test_data = [['amEricA',	'1010101'], ['123abc',	'000100'], ['Hello, world!',	'0100100010000'],
             ['ＡｂＣＤｅ', '10001'], ['   !@# ,.', '000000000'] ]

for s, exp in test_data:
    t = re.sub(r'.', lambda m: '01'[m[0] in 'aeiouAEIOUａｅｉｏｕＡＥＩＯＵ'], s)
    print(t, t == exp)

Comments

[Cocode]

回答ありがとうございますm(_ _)m
そして質問失礼します！

'01'[m[0] in 'aeiouAEIOUａｅｉｏｕＡＥＩＯＵ']
↑この部分が理解できませんでした…

①m[0]とはどういう意味ですか？
②m[0] in 'aeiouAEIOUａｅｉｏｕＡＥＩＯＵ'　←この部分はなんとなく理解できるのですが、
'01'[...]　←こういう形を始めて拝見しました（文字列と配列がくっついている）。どういう動作でしょうか？

[退会済みユーザー]

m[0]は、正規表現でマッチさせた文字です。
マッチオブジェクトの説明(https://docs.python.org/ja/3/library/re.html#match-objects)の
Match.group([group1, ...]), Match.__getitem__(g) を読んでみてください。
このプログラムでは、m.group(), m.group(0), m[0] どれでも同じです。

m[0] in 'aeiou...' で取り出した文字が母音かどうかを判定しTrue/Falseにします。
pythonではTrue==1, False==0なので、文字列 '01' のインデックスとして使用できます。
'01'[False]は'0'になり、'01'[True]は'1'になります。

Answer 3

re.subには、置換対象として関数を渡すことができます（Python公式）。

Comments

[Cocode]

迅速な回答ありがとうございます！