Python Return int from Enum class

Question

The aim is to use an Enum class to allow for predictable input of a definite amount of options.

I expected to get the integers as output, however I get class.attrib if I print it.

How can I get the integer instead?

from enum import Enum

class Stiffness(Enum):
    FREE = 0
    RIGID = 1
    FLEXIBLE = 2

print(Stiffness.FLEXIBLE)

Output: Stiffness.FLEXIBLE

Expected output: 2

Thierry Lathuille · Accepted Answer · 2020-12-08 11:39:36Z

8

Just print the value:

print(Stiffness.FLEXIBLE.value)
# 2

answered Dec 8, 2020 at 11:39

Thierry Lathuille

24.4k10 gold badges49 silver badges57 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

mkrieger1 · Accepted Answer · 2020-12-08 13:44:29Z

4

You can use IntEnum instead of Enum.

This allows to use Stiffness.FLEXIBLE interchangeably with the integer 2, for example you can pass it to a function that expects an integer.

answered Dec 8, 2020 at 11:42

mkrieger1

24.2k7 gold badges68 silver badges84 bronze badges