Accès le seul élément d'un ensemble

J'ai un set en Python à partir de laquelle je suis supprimer les éléments un par un en fonction d'une condition. Quand l'ensemble est laissé avec seulement 1 élément, j'ai besoin de retour de cet élément. Comment puis-je accéder à cet élément de l'ensemble?

Un exemple simplifié:

S = set(range(5))
for i in range(4):
    S = S - {i}
# now S has only 1 element: 4
return ? # how should I access this element
# a lame way is the following
# for e in S:
#    return S
  • Si vous voulez supprimer un élément d'un ensemble, S.remove(i) est plus rapide. (Notez que S.remove(i) modifie S, tandis que S = S - {i} le remplace par un nouvel ensemble.)
InformationsquelleAutor Nik | 2013-12-17
  1. 27

    Utilisation set.pop:

    >>> {1}.pop()
1
>>>

    Dans votre cas, il serait:

    return S.pop()

    Note toutefois que cela va supprimer l'élément de l'ensemble. Si cela n'est pas souhaitable, vous pouvez utiliser min|max:

    return min(S) # 'max' would also work here

    Démo:

    >>> S = {1}
>>> min(S)
1
>>> S
set([1])
>>> max(S)
1
>>> S
set([1])
>>>
    • +1, tu ne connaissais pas set.pop bien qu'il n'est pas trop surprenant, je suppose. Et ici, je l'aurais fait next(iter(S))
    • Seulement si vous êtes d'accord avec la suppression de l'élément. Sinon, un for boucle ou next(iter(S)) fonctionne mieux.
    • Off the record, j'ai tendance à utiliser min(S), bien que j'ai Officiellement en charge(tm) next(iter(S)).
    • En fait, je l'aime mieux que l'autre méthode. Il est plus court.
    • À partir d'une documentation du code perspective, max() ou min() sont juste ... mal. Trop surprenant de la méfiance du lecteur de code.
    InformationsquelleAutor iCodez
  2. 27

    Je voudrais utiliser:

    e = next(iter(S))

    C'est non destructive, et qui fonctionne même quand il n'y a plus d'un élément dans l'ensemble. Encore mieux, il a une option pour fournir une valeur par défaut e = next(iter(S), default).

    Vous pouvez également utiliser le déballage:

    [e] = S

    Le déballage technique est susceptible d'être le moyen le plus rapide et il comprend la vérification des erreurs pour assurez-vous que le jeu n'a qu'un seul membre. L'inconvénient est qu'il fait un peu bizarre.

    • Je timeit d et le déballage semble être de 2,5 x plus rapide que la voie d'iter. Et min/max sont environ 2 fois plus rapide que iter trop.
    • Le déballage approche d'affectation est sympa aussi.
    • Ou si vous n'êtes pas friands de la virgule, des crochets sera trop de travail: [e] = S.
    • Le déballage a l'air un peu bizarre, mais des options semble le moins mauvais.
    InformationsquelleAutor Raymond Hettinger
  3. 2

    Désolé, la fin de la partie. Pour accéder à un élément d'un ensemble, vous pouvez toujours lancer le jeu dans une liste et vous pouvez alors utiliser l'indexation de retourner la valeur que vous souhaitez.

    Dans le cas de votre exemple:

    return list(S)[0]
    InformationsquelleAutor J4y
  4. 0

    Vous pouvez affecter le dernier élément d'une variable en utilisant star operation.

    >>> a={"fatih"}
>>> b=str(*a)
>>> b
'fatih'

    Maintenant la varible b ont un string object.

    Si la seule elemant est un nombre, vous pouvez utiliser int():

    >>> a={1}
>>> c=int(*a)
>>> c
1
    InformationsquelleAutor Fatih1923