実現したい事

フォーマットや深さが一定でないJSON形式のデータから、キーワードを指定しその配下を抜き出す関数を実装したい
一致するキーワードが見つかった場合、再帰処理自体を終了したい

不明点

return文で実装したのですが、キーワードに一致した場合も後続の再帰処理が続いてしまうため、関数の処理自体を終了する方法をご教授下さい

Python
1def extract_json(json_data, target_words):
2    """
3    指定した項目名をJSONから抜き出す
4    """
5    
6    if isinstance(json_data, dict):
7        for key in json_data.keys():
8            if json_data[key] is None:
9                continue
10            if key == target_words:
11                    return json_data[key]  # ★★★キーワードに一致した場合は、ここで関数処理自体を終了したい                    
12            extract_json(json_data[key], target_words)
13
14    elif isinstance(json_data, list):
15        for item in json_data:
16                extract_json(item, target_words)
17
18# メイン処理
19extract_json(json_data, "target_word")

行動規範の内容に同意します

回答3件

戻り値かなんかで終了の値を返し、その値が返されたらリターンするようにする、とかでどうでしょう。

投稿2021/03/08 07:21

y_waiwai

総合スコア88038

python
1            if json_data[key] is None:
2                continue

としてるのだからNoneでreturnする気はない。ということだと思います。
また、returnせずに終了するとNoneが返る。ということは
None以外が返ってきたら見つかったということで、returnすればいいのでは？

具体的には以下で良いんじゃないでしょうか。

python
1def extract_json(json_data, target_words):
2    """
3    指定した項目名をJSONから抜き出す
4    """
5
6    if isinstance(json_data, dict):
7        for key in json_data.keys():
8            if json_data[key] is None:
9                continue
10            if key == target_words:
11                    return json_data[key]  # ★★★キーワードに一致した場合は、ここで関数処理自体を終了したい                    
12            rtn = extract_json(json_data[key], target_words) 
13            if not rtn is None:
14                return rtn
15
16    elif isinstance(json_data, list):
17        for item in json_data:
18                rtn = extract_json(item, target_words)
19                if not rtn is None:
20                    return rtn
21
22
23# メイン処理
24extract_json(json_data, "target_word")

投稿2021/03/08 14:11

xail2222

総合スコア1508

こういう再帰関数からの大域脱出は例外処理を使うとスマートです。

手抜きでやると、

python
1def extract_json(json_data, target_words):
2    if isinstance(json_data, dict):
3        for key in json_data.keys():
4            if json_data[key] is None:
5                continue
6            if key == target_words:
7                excp = Exception()
8                excp.value = json_data[key] 
9                raise excp
10            extract_json(json_data[key], target_words)
11    elif isinstance(json_data, list):
12        for item in json_data:
13            extract_json(item, target_words)
14
15def extract_json_top(json_data, target_words):
16    try:
17        extract_json(json_data, target_words)
18        return None
19    except Exception as e:
20        return e.value
21
22# メイン処理
23extract_json_top(json_data, "target_word")

実行結果

python
1>>> data ={'foo':42, 'bar':[{'baz':420, 'qux':4200}]}
2>>> print(extract_json_top(data, 'aaa'))
3None
4>>> print(extract_json_top(data, 'baz'))
5420