pythonの文字列リストから数字を取り出す方法

Question

python初心者です。
pythonのリスト['a=1 b=2 c=3', 'a=2 b=3, c=4']となっているときにaの値のみを別のリストに入れるにはどうしたら良いでしょうか。
具体的には[1, 2]という出力が欲しいです。

Answer 1

こんにちは

a の値を得るために、以下の手順を考えました。

1. 'a=1 b=2 c=3'という文字列から { 'a':1, 'b':2, 'c':3 } というdictを得る関数 get_dict を作っておき、
2. この関数によって、所与の文字列のリスト text から、dictのリスト dicts を作り、
3. dicts から 'a' の値のリストを得る

以下、上記をコードにしたものです。

import re


def get_dict(params_str):
    params = re.split(r',?\s+', params_str)
    return dict([re.split(r'=', param) for param in params])


if __name__=='__main__':
    text = ['a=1 b=2 c=3', 'a=2 b=3, c=4']
    dicts = [get_dict(params_str) for params_str in text]
    a_values = [int(d['a']) for d in dicts]

    print(a_values)  # => [1, 2]

• 動作確認用 Repl.it: https://repl.it/@jun68ykt/Q250113

上記の get_dictは、ご質問にある text の要素のパターンから正規表現を作っているので、微妙なところでご質問とは異なるパターンが text の要素に出現すると対応できなくなりますが、その都度、get_dict で使っている正規表現を見直すなどして対応していけばよいかと思います。

以上参考になれば幸いです。

追記

別の方法を挙げておきます。'a=1 b=2 c=3' という文字列に含まれるスペースを &で置き換えると、
'a=1&b=2&c=3'
が得られますが、これはURLの一部を構成するクエリ文字列の形式なので、これをパースするモジュールの力を借りることができます。以下は、この考え方によって、関数get_dict を修正したコードです。

from urllib.parse import parse_qs 


def get_dict(params_str):
    qs = params_str.replace(' ', '&')
    return parse_qs(qs)

if __name__=='__main__':
    text = ['a=1 b=2 c=3', 'a=2 b=3 c=4']
    dicts = [get_dict(params_str) for params_str in text]
    a_list = [int(d['a'][0]) for d in dicts]
    print(a_list)

• 動作確認用 Repl.it: https://repl.it/@jun68ykt/Q2501132

• 参考: urllib.parse.parse_qs

Comments

[edamame11]

とても参考になりました。ありがとうございます。

[jun68ykt]

@edamame11さん
どういたしまして。追記に書いた方法が採用できると、さらに楽かもしれません。

Answer 2

['a=1 b=2 c=3', 'a=2 b=3, c=4']

ここの仕様（たとえば、このリストには　'b=1 a=3 c=2' もくるのかとか、 'a=3.2 b=1 c=3' もくるのかとか）が明記されていないのでなんとも回答しにくいのですが、1行で書くとこんな感じです。

import re
[int(re.match(r'(^|.* )a=(\d*)($| .*)',item).group(2)) for item in ['a=1 b=2 c=3', 'a=2 b=3, c=4']]

利用している技術は以下のとおりです。

• 正規表現で文字列から対象部分の切り出し
• int 関数で文字列から数値に変換
• リスト内包表記で個々の要素に関数を適用した結果のリストの生成

Comments

[edamame11]

ご回答していただきありがとうございます。
もう少し詳しく明記すればよかったですね。申し訳ありません。

abc_list = ['a=1 b=2 c=3', 'a=2 b=3 c=4'....]
(abcにはそれぞれ不規則な数字が入る、abcの順番は固定、要素数が2つ以上ある)場合
for文でappendを使って、a_list = []にaの値を入れることは可能でしょうか？

[mit0223]

> abcにはそれぞれ不規則な数字が入る、abcの順番は固定、要素数が2つ以上ある
この仕様であれば、回答に示したサンプルで処理できそうです。

> for文でappendを使って、a_list = []にaの値を入れることは可能でしょうか？
もちろん、for文、append でもできますが、リストの内包表記も覚えておいて損はないと思います。