重载列表理解行为？

Question

我的任务是创建一个硬件笼模型。每个笼子包含 N 个插槽，每个插槽可能包含也可能不包含卡片。

我想使用列表来模拟笼子。每个列表索引将对应于槽号。 cards[0].name="Card 0" 等。

这将允许我的用户通过简单的列表理解来查询模型。例如：

for card in cards:
    print card.name

我的用户不是熟练的 Python 用户，他们将与模型实时交互，因此让列表索引不对应于填充的卡片是不切实际的。换句话说，如果用户移除一张卡，我需要执行一些操作来指示该卡未填充 - 我的第一反应是将列表项设置为 None。

老板喜欢这个方案，但他并不热衷于上面的列表理解如果缺少一张卡片就会失败。 （目前确实如此。）他甚至不太支持要求用户学习足够的 Python 来创建忽略 None 的列表理解表达式。

我的想法是对 list 类进行子类化，以创建一个 newclass。它的工作方式与列表完全相同，但 for card in cards 只会返回未设置为 None 的成员。

有人可以演示如何重载列表类，以便子类上调用的列表推导式将忽略 None 吗？ （到目前为止，当我尝试这样做时，我的 Python 技能已经开始下降。）

任何人都可以提出更好的方法吗？

Answer 1

您可以为此提供一个生成器/迭代器。

def installed(cage):
    for card in cage:
        if card:
            yield card

cards = ["Adaptec RAID", "Intel RAID", None, "Illudium Q-36 Explosive Space Modulator"]

# print list of cards
for card in installed(cards):
    print card

Answer 2

如果您使用的是 2.6 或更高版本，您可以执行类似的操作来获取名称：

names = [x.name for x in cards if x is not None]

我认为这应该接近您想要的。

Answer 3

也许定义一个函数（假设cards是一个全局变量？！？）：

def pcards():
    for card in cards:
        if card:
            print card.name

这样你的用户就可以简单地输入pcards()来获取列表。

Answer 4

>>> class MyList(list):
...     def __iter__(self):
...         return (x for x in list.__iter__(self) if x is not None)
... 
>>> 
>>> ml = MyList(["cat", "dog", None, "fox"])
>>> for item in ml:
...     print item
... 
cat
dog
fox

>>> [x for x in ml]
['cat', 'dog', 'fox']
>>> list(ml)
['cat', 'dog', 'fox']