python - python - リスト内の辞書内の要素を検索する

Question

私のデータ構造は次のようになります。

  - testSet: a list of records in the test set, where each record
             is a dictionary containing values for each attribute

そして、各レコードには「ID」という名前の要素があります。testSetID値で内部のレコードを検索したいと思います。したがって、ID = 230 が与えられた場合、ID 要素が 230 に等しいというレコードを返したいと思います。

どうやってやるの？

score 0 · Accepted Answer

例えば：

set = [{'ID': 50}, {'ID': 80}]

def find_set(id):
    return [elem for elem in set if elem['ID'] == id]

これにより、指定された ID を持つすべてのアイテムが返されます。最初のものだけが必要な場合は、[0] を追加します (存在するかどうかを確認した後、例:

def find_set(id):
    elems = [elem for elem in set if elem['ID'] == id]
    return elems[0] if elems else None

3 に答える 3