Pythonでオブジェクトのリストを検索する

データ要素を保持するために、Cスタイルの構造体と同様に機能する単純なクラスを作成しているとしましょう。特定の値に等しい属性を持つオブジェクトのオブジェクトのリストを検索する方法を理解しようとしています。以下は、私が何をしようとしているのかを説明する簡単な例です。

例えば：

class Data:
    pass

myList = []

for i in range(20):
    data = Data()
    data.n = i
    data.n_squared = i * i
    myList.append(data)

myListリストを検索して、n == 5の要素が含まれているかどうかを確認するにはどうすればよいですか？

私はPythonドキュメントをグーグル検索して検索してきましたが、これをリスト内包表記で実行できると思うかもしれませんが、よくわかりません。ちなみに、Python 2.4.3を使用する必要があるため、新しいgee-whiz 2.6または3.xの機能は利用できません。

リスト内包表記を使用して、一致するすべての要素のリストを取得できます。

[x for x in myList if x.n == 30]  # list of all elements with .n==30

リストに一致する要素が含まれているかどうかを判断し、それを（比較的）効率的に行う場合は、次のようにすることができます。

def contains(list, filter):
    for x in list:
        if filter(x):
            return True
    return False

if contains(myList, lambda x: x.n == 3)  # True if any element has .n==3
    # do stuff

シンプル、エレガント、パワフル：

ビルトインと組み合わせたジェネレータ式…（python 2.5+）

any(x for x in mylist if x.n == 10)

any()次のように定義されているPython 組み込みを使用します。

any（iterable）イテラブルの -> いずれかの要素がtrueの場合、trueを返します。に相当：

def any(iterable):
    for element in iterable:
        if element:
            return True
    return False

完全を期すために、おそらく機能する最も単純なものを忘れないでください。

for i in list:
  if i.n == 5:
     # do something with it
     print "YAY! Found one!"

[x for x in myList if x.n == 30]               # list of all matches
[x.n_squared for x in myList if x.n == 30]     # property of matches
any(x.n == 30 for x in myList)                 # if there is any matches
[i for i,x in enumerate(myList) if x.n == 30]  # indices of all matches

def first(iterable, default=None):
  for item in iterable:
    return item
  return default

first(x for x in myList if x.n == 30)          # the first match, if any

filter(lambda x: x.n == 5, myList)

を使用inして、コレクション内のアイテムを検索し、リスト内包表記を使用して、目的のフィールドを抽出できます。これは、リスト、セット、タプル、__contains__またはを定義するものすべてで機能します__getitem__。

if 5 in [data.n for data in myList]:
    print "Found it"

以下も参照してください。

• メソッドを含む
• 稼働中

クラスにメソッド__eq__と__hash__メソッドを追加する必要があります。Data__dict__属性が等しい（同じプロパティ）かどうか、そしてそれらの値も等しい。

あなたがそれをしたなら、あなたは使うことができます

test = Data()
test.n = 5

found = test in myList

in場合は、キーワードをチェックしtestていますmyList。

nプロパティのみをData使用する場合は、次のように使用できます。

class Data(object):
    __slots__ = ['n']
    def __init__(self, n):
        self.n = n
    def __eq__(self, other):
        if not isinstance(other, Data):
            return False
        if self.n != other.n:
            return False
        return True
    def __hash__(self):
        return self.n

    myList = [ Data(1), Data(2), Data(3) ]
    Data(2) in myList  #==> True
    Data(5) in myList  #==> False

辞書の使用を検討してください：

myDict = {}

for i in range(20):
    myDict[i] = i * i

print(5 in myDict)

もう1つの方法は、next（）関数を使用することです。

matched_obj = next(x for x in list if x.n == 10)