配列の中のハッシュのキーに特定の文字列が含まれる要素の値を取り出したい

Question

前提・実現したいこと

fooの値である'2020-01-01'をスマートに取得するには、Python3ではどのように記述すると良いでしょうか？
なお、1つ目の要素にあるとは限りません

d = {'start_date': [{'foo': '2020-01-01'}, {'bar': '2020-01-02'}]}

以下のようにそもそもfooやbarが多次元のHashに入っていればもう少しやりようがあると思います。
しかし、この元データ構造は変えられない前提です。（YAMLから自動生成されているようです）

d2 = {'start_date': {'foo': '2020-01-01', 'bar': '2020-01-02'}}

何卒よろしくお願いいたします。

試したこと

items()を用いてfilter関数を使おうとしましたが、配列にはばかられ、値が取れませんでした

>>> list(filter(lambda item: item == 'foo', d)
[]

補足情報（FW/ツールのバージョンなど）

Python3です。pandasを使って解決する手法も歓迎です

Answer 1

filter を使うと

s = dict(*filter(lambda x: 'foo' in x , d['start_date']))['foo']
print(s)  # 2020-01-01

辞書内包表記を使うと（modieu さんの別の書き方）

s = {k: chd[k] for chd in d['start_date'] for k in chd}['foo']
print(s)  # 2020-01-01

Comments

[y-ken]

動作するコード、ありがとうございます！
filterで取り出したいと思っていました。このように記述するのですね。試してみます

Answer 2

スマートではありませんが、辞書とリストが何重になっていても見つけることの出来る関数です。

def search_dl(dl, key):
    if type(dl) == list:
        return sum([search_dl(x, key) for x in dl], [])
    elif type(dl) == dict:
        if (key in dl) and type(dl[key]) == str:
            return [dl[key]] + sum([search_dl(dl[x], key) for x in dl], [])
        else:
            return sum([search_dl(dl[x], key) for x in dl], [])
    else:
            return []

実行結果

>>> d = {'start_date': [{'baz': '2020-01-01'}, {'bar': '2020-01-02'}], 'end_date': [{'corge': '2020-01-03'}, {'foo': '2020-01-04'}, {'qux': [{'grault': '2020-01-05'}, {'quux': '2020-01-06'}]}]}
>>>
>>> search_dl(d, 'foo')
['2020-01-04']
>>> search_dl(d, 'quux')
['2020-01-06']

Comments

[y-ken]

動作するコード、ありがとうございます！
試してみます

[y-ken]

とても汎用的なので、手元のutil.pyに追加しようと思います

Answer 3

少し冗長な感じで書くと、やりたいことは以下のような処理でいいでしょうか？

>>> def get_foo_start_date():
...     d={'start_date': [{'foo': '2020-01-01'}, {'bar': '2020-01-02'}]}
...     for x in d["start_date"]:
...         if x.get("foo", None):
...             return x["foo"]
...
>>>
>>> get_foo_start_date()
'2020-01-01'

ワンライナーを使って似たような処理だと、以下のやり方でどうでしょうか？

>>> d = {'start_date': [{'foo': '2020-01-01'}, {'bar': '2020-01-02'}]}
>>> [ x["foo"] for x in d["start_date"] if "foo" in x ]
['2020-01-01']
>>>
>>>
>>> [ x["foo"] for y in d.values() for x in y if "foo" in x ]
['2020-01-01']

Comments

[y-ken]

動作するコード、ありがとうございます！
試してみます

Answer 4

{k: v for chd in d['start_date'] for k, v in chd.items()}['foo']

とか

Comments

[y-ken]

動作するコード、ありがとうございます！
試してみます