2

我有一些代码从列表中返回值。我正在使用强类型遗传编程(使用优秀的 DEAP 模块),但我意识到1& 与and0相同。这意味着一个函数需要一个整数,它可能会以一个布尔函数结束,这会导致一些问题。TrueFalse

例如: list = [1,2,3,4,5]

list[1]返回2

list[True]也返回2

有没有一种 Pythonic 的方法来防止这种情况?

4

1 回答 1

3

您可以定义自己的不允许布尔索引的列表:

class MyList(list):
    def __getitem__(self, item):
        if isinstance(item, bool):
            raise TypeError('Index can only be an integer got a bool.')
        # in Python 3 use the shorter: super().__getitem__(item)
        return super(MyList, self).__getitem__(item)

做一个实例:

>>> L = MyList([1, 2, 3])

整数有效:

>>> L[1]
2

True不:

>>> L1[True]
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-888-eab8e534ac87> in <module>()
----> 1 L1[True]

<ipython-input-876-2c7120e7790b> in __getitem__(self, item)
      2     def __getitem__(self, item):
      3         if isinstance(item, bool):
----> 4             raise TypeError('Index can only be an integer got a bool.')

TypeError: Index can only be an integer got a bool.

相应地覆盖__setitem__以防止使用布尔值作为索引设置值。

于 2015-11-15T10:47:18.993 回答