如何在每次迭代中获取 Enum 属性的随机值？

Question

我已经创建了这样的Enum 对象：

class Gender(Enum):
    FEMALE = 'female'
    MALE = 'male'
    RANDOM = random.choice([FEMALE, MALE])

我每次都想得到真正随机的值，但它不起作用：

>>> class Gender(Enum):
...    MALE = 'male'
...    FEMALE = 'female'
...    RANDOM = choice([MALE, FEMALE])
... 
>>> Gender.RANDOM
<Gender.MALE: 'male'>
>>> Gender.RANDOM
<Gender.MALE: 'male'>
>>> Gender.RANDOM
<Gender.MALE: 'male'>
>>> Gender.RANDOM
<Gender.MALE: 'male'>

我也尝试过使用 lambda，但它看起来不太好，虽然它可以工作：

Gender.RANDOM()

有没有其他方法可以在不使用 lambda 表达式的情况下每次获取随机值？

我们使用这个枚举对象作为一些方法的参数的默认值，这就是为什么它应该是一个属性，而不是一个函数，因为当我们使用Gender.FEMALE它不是一个函数，它是一个属性和Gender.RANDOM也应该是一个属性：

def full_name(gender=Gender.FEMALE):
    ...


def full_name(gender=Gender.RANDOM):
    ...

Answer 1

我认为你想要一个随机的方法，而不是直接将值保存在变量中，因为如果你这样做，值将永远不会改变：

如果您不想要函数

随机导入 导入枚举

class Gender(enum.Enum):

  MALE = 'male'
  FEMALE = 'female'
  RANDOM = random.choice([MALE, FEMALE])


  def __getattribute__(self,item):
    if item == "RANDOM":
      Gender.RANDOM = random.choice([self.MALE, self.FEMALE])
      return Gender.RANDOM
    else:
      return object.__getattribute__(self, item)

gender = Gender()

看：

   gender.RANDOM
=> 'female'
   gender.RANDOM
=> 'male'
   gender.RANDOM
=> 'male'
   gender.RANDOM
=> 'male'
   gender.RANDOM
=> 'female'
   gender.RANDOM
=> 'male'
   gender.RANDOM
=> 'male'
   gender.RANDOM
=> 'female'

Answer 2

您可能应该在Enum 中创建一个方法来获取随机性别：

import random
import enum

class Gender(enum.Enum):
    FEMALE = 'female'
    MALE = 'male'

    @classmethod
    def get_gender(cls):
        return random.choice([Gender.FEMALE, Gender.MALE])

Gender.get_gender()

Answer 3

来自文档：“枚举是一组符号名称（成员），绑定到唯一的常量值。” 您需要切换到另一种表示形式，例如类。

Answer 4

我尝试了metaclasses 的方法。它有效！

import random
import enum
class RANDOM_ATTR(enum.EnumMeta):
    @property
    def RANDOM(self):
        return random.choice([Gender.MALE, Gender.FEMALE])


class Gender(enum.Enum,metaclass=RANDOM_ATTR): #this syntax works for python3 only
    FEMALE = 'female'
    MALE = 'male'


print(Gender.RANDOM)   #prints male or female randomly

这里通过使RANDOM_ATTR 成为Gender 的元类，Gender 就像类RANDOM_ATTR 的对象，所以Gender 具有property RANDOM。

但是，您在问题中描述的以下代码无法按您预期的方式工作。

def full_name(gender=Gender.RANDOM):
    ...

RANDOM 属性只会被调用一次。要了解原因，请阅读此answer。默认参数就像函数的属性，只会被初始化一次。

为此，我建议您这样做：

def full_name(gender=None):
    gender = gender or Gender.RANDOM
    ...

Answer 5

尽管在许多情况下修改类可能会更清晰和 DRY-er，但如果您只这样做一次或不想修改枚举，您可以这样做：

random.choice([enm.value for enm in Gender])

Answer 6

由于RANDOM 不是真正枚举中的项目，我认为更连贯的方法是将其精确地保留为方法而不是属性（毕竟不是！）。

import random
import enum


class Gender(enum.Enum):
    MALE = 'male'
    FEMALE = 'female'

    @staticmethod
    def random():
        return random.choice(list(Gender))

然后，您可以将“我没有选择”行为转移到实际上更有意义的函数中。

def full_name(gender=None):
    if gender is None:
        gender = Gender.random()
    # ...

Answer 7

正如其他人所说，最好的方法是让 random() 成为枚举类上的一个方法，以明确 RANDOM 不是成员。

但是，因为我喜欢谜题：

from enum import Enum
import random

class enumproperty(object):
    "like property, but on an enum class"

    def __init__(self, fget):
        self.fget = fget

    def __get__(self, instance, ownerclass=None):
        if ownerclass is None:
            ownerclass = instance.__class__
        return self.fget(ownerclass)

    def __set__(self, instance, value):
        raise AttributeError("can't set pseudo-member %r" % self.name)

    def __delete__(self, instance):
        raise AttributeError("can't delete pseudo-member %r" % self.name)

class Gender(Enum):
    FEMALE = 'female'
    MALE = 'male'
    @enumproperty
    def RANDOM(cls):
        return random.choice(list(cls.__members__.values()))

---

在您的full_name 定义中，使用Gender.RANDOM 作为默认值不会得到您想要的。这样的标准是：

def full_name(gender=None):
    if gender is None:
        gender = Gender.RANDOM   # we get `MALE` or `FEMALE`, not `RANDOM`

这会让读者感到困惑。使用普通方法会更好：

def full_name(gender=None):
    if gender is None:
        gender = Gender.random()

Answer 8

另一个适合您目标的选项是：

from enum import Enum
import random

class Gender(Enum):
    MALE = object()
    FEMALE = object()

    @classmethod
    def get_random_gender(cls):
        gender = random.choice(cls._member_names_)
        return gender.lower()


print(Gender.get_random_gender())
>> female