Wie konvertiert man Int in Enum in Python?

Mithilfe der neuen Enum-Funktion (über backport enum34) mit python 2.7.6.

Gegeben, die folgende definition, wie konvertiere ich ein int auf die entsprechende Enum-Wert?

from enum import Enum

class Fruit(Enum):
    Apple = 4
    Orange = 5
    Pear = 6

Ich weiß, ich kann die hand-Handwerk eine Reihe von if-Anweisungen, um die Konvertierung zu tun, aber gibt es eine einfache pythonic Weg zu konvertieren? Im Grunde möchte ich eine Funktion ConvertIntToFruit(int) liefert einen enum-Wert.

Mein use-case ist, ich habe eine csv Datei mit Datensätzen, wo lese ich jeden Datensatz in ein Objekt. Eines der Datei Felder ist ein integer-Feld stellt eine Aufzählung. Da bin ich Auffüllen der Gegenstand den ich umwandeln möchte, das integer-Feld aus der Datei in den entsprechenden Enum-Wert in das Objekt.

InformationsquelleAutor der Frage User | 2014-05-30

  1. 74

    'Call' die Enum Klasse:

    Fruit(5)

    drehen 5 in Fruit.Orange:

    >>> from enum import Enum
>>> class Fruit(Enum):
...     Apple = 4
...     Orange = 5
...     Pear = 6
... 
>>> Fruit(5)
<Fruit.Orange: 5>

    Aus der Den programmatischen Zugriff auf die Aufzählung der Mitglieder und deren Attribute Abschnitt der Dokumentation:

    Manchmal ist es sinnvoll, den Zugriff auf Mitglieder in Aufzählungen
    programmatisch (D. H. Situationen, in denen Color.red nicht tun, weil die
    genau die Farbe, ist nicht bekannt-Programm-schreiben-Zeit). Enum wie
    Zugang:

    >>> Color(1)
<Color.red: 1>
>>> Color(3)
<Color.blue: 3>

    InformationsquelleAutor der Antwort Martijn Pieters

  2. -3

    Können Sie auch python-dictionary für diesen Zweck. Wie folgt,

    Fruit = {4:'Apple',5:'Orange',6:'Pear'}
Fruit[4]
Fruit[6]

    Ausgabe:

    Apple
Pear

    Oder Sie erstellen ein einfaches Enum wie folgt,

    Apple,Orange,Pear = range(4,6)
print Apple,Pear,Orange

    InformationsquelleAutor der Antwort Naggappan Ramukannan

Schreibe einen Kommentar