リストから条件に一致するデータだけ新しくリストにする

Question

リストから条件に一致するデータだけ新しくリストにする書き方がわかりません。

以下のリストがあります。

q = [
    {
        "id": 1,
        "fruits": "apple",
        "cd": [1,2,3]
    },
    {
        "id": 2,
        "fruits": "banana",
        "cd": [1]
    },
    {
        "id": 3,
        "fruits": "grape",
        "cd": [2,3]
    }
]

入力されたリストの値が cd のリストに含まれていることを条件にして出力内容を変更したいです。

input = [1] のときの出力内容

r = [
    {
        "id": 1,
        "fruits": "apple",
        "cd": [1,2,3]
    },
    {
        "id": 2,
        "fruits": "banana",
        "cd": [1]
    }
]

input = [1,3] のときの出力内容

r = [
    {
        "id": 1,
        "fruits": "apple",
        "cd": [1,2,3]
    }
]

input = [2,3] のときの出力内容

r = [
    {
        "id": 1,
        "fruits": "apple",
        "cd": [1,2,3]
    },
    {
        "id": 3,
        "fruits": "grape",
        "cd": [2,3]
    }
]

作成したコード

r = []
input = [2,3]     # input = [2,3]の場合

for item in q:
        if input[0] in item["cd"] and input[1] in item["cd"]:
            r.append(item)
print(r)

inputの条件の個数に対して動的に書くことができていません。要素の個数に関係なく出力する書き方を教えていただきたいです。

Answer 1

can110さんの回答と同様にsetに変換しinputの要素が各cdに含まれるかの判定に <= を使う案です：

r = [e for e in q if {*input} <= {*e['cd']}]

a と b がsetである場合に、a <= b は a が b の部分集合である場合に True となり、そうでない場合に False になります。

Comments

[sasaki0628]

回答ありがとうございます。シンプルでわかりやすいです。

Answer 2

q = [
    {"id": 1, "fruits": "apple",  "cd": [1,2,3]},
    {"id": 2, "fruits": "banana", "cd": [1]},
    {"id": 3, "fruits": "grape",  "cd": [2,3]}
]

for inp in ([1], [1, 3], [2, 3]):
    r = [*filter(lambda x: all(i in x['cd'] for i in inp), q)]
    print(f'{inp} => {r}')

# [1] => [{'id': 1, 'fruits': 'apple', 'cd': [1, 2, 3]}, {'id': 2, 'fruits': 'banana', 'cd': [1]}]
# [1, 3] => [{'id': 1, 'fruits': 'apple', 'cd': [1, 2, 3]}]
# [2, 3] => [{'id': 1, 'fruits': 'apple', 'cd': [1, 2, 3]}, {'id': 3, 'fruits': 'grape', 'cd': [2, 3]}]

Comments

[sasaki0628]

回答ありがとうございます。filterを知りませんでした。勉強になります。

[sasaki0628]

時間が経ってから見直したときにこちらの書き方が一番わかりやすく感じました。ありがとうございました。

Answer 3

どちらも集合(set)として扱うことで、入力から引いた結果が空集合＝入力されたすべての要素を含む と判定できます。

q = [{'id': 1, 'fruits': 'apple', 'cd': [1, 2, 3]}, 
     {'id': 2, 'fruits': 'banana', 'cd': [1]}, 
     {'id': 3, 'fruits': 'grape', 'cd': [2, 3]}]
for inp in [[1],[1,3],[2,3]]:
    r = [i for i in q if len(set(inp) - set(i['cd'])) <= 0 ]
    print(inp,r)

# [1] [{'id': 1, 'fruits': 'apple', 'cd': [1, 2, 3]}, {'id': 2, 'fruits': 'banana', 'cd': [1]}]
# [1, 3] [{'id': 1, 'fruits': 'apple', 'cd': [1, 2, 3]}]
# [2, 3] [{'id': 1, 'fruits': 'apple', 'cd': [1, 2, 3]}, {'id': 3, 'fruits': 'grape', 'cd': [2, 3]}]

Comments

[sasaki0628]

回答ありがとうございます。型を集合にして計算するとうまくいくんですね。