如何从列表中的列表中删除对象？

Question

假设有三个列表，第三个是包含前两个的列表：

Friendly_List = list()
Enemy_List = list()
Battle_Space = list([Friendly_List, Enemy_List])

还有一个地精类：

class Goblin(object):
    def __init__(self):
        self.Name = "Goblin"
        self.HP = random.randint(15,20)
        self.Damage = random.randint(5,10)

我创建了一个 Goblin 类的实例：

x = Goblin()

然后放入Friendly_List：

Friendly_List.append(x)

现在假设地精英勇地死去，必须从战场上移除。如果我执行 Friendly_List.remove(x)，则可以毫无问题地移除地精。

如果我执行 Battle_Space.remove(x)，它表示 x 不在列表中 - 即使 x 在属于 Battle_Space 列表的列表之一中。

我想我真正想问的是：实现 Battle_Space.remove(x) 所需行为的正确方法是什么？移除一个深层的实例？

Answer 1

简单但可能不太干净的方法是Battle_Space[0].remove(x)。您必须确保第一个列表是友好列表。

我对可读性的建议是考虑将friendly_list 和enemy_list 放在BattleSpace 对象中，以便您可以通过名称引用它们。您还可以在该对象中创建一个方法以从任何子列表中删除目标：

class BattleSpace(object):
    def __init__(self):
        self.friendly_list = []
        self.enemy_list = []

    def remove(self, target):
        if target in self.friendly_list:
            self.friendly_list.remove(target)
        if target in self.enemy_list:
            self.enemy_list.remove(target)


battle_space = BattleSpace()
battle_space.friendly_list.append(x)
battle_space.remove(x)  # can also do battle_space.friendly_list.remove(x)

编辑：只需阅读 OP 的评论，即希望删除列表中的任何位置。修改后的代码。

Answer 2

如果知道嵌套列表的索引，可以从父列表索引，调用remove：

Battle_Space[0].remove(x)

要从它出现的任何地方删除x，您可以使用列表推导：

Battle_Space = [[j for j in innerlist if j!=x] for innerlist in Battle_Space]

Answer 3

只需过滤掉那些该死的地精，例如：

import random


class Goblin(object):

    def __init__(self):
        self.Name = "Goblin"
        self.HP = random.randint(15, 20)
        self.Damage = random.randint(5, 10)

x = Goblin()
Friendly_List = list()
Enemy_List = list()
Battle_Space = list([Friendly_List, Enemy_List])
Friendly_List.append(x)
Friendly_List.append(x)
Friendly_List.append(x)
Enemy_List.append(x)

print Battle_Space
for index, army in enumerate(Battle_Space):
    Battle_Space[index] = filter(lambda entity: entity != x, army)
print Battle_Space

代码将从你的星际战斗的所有军队中移除该死的复制地精。

我不是银河战斗的忠实粉丝，地精可以神奇地复制自己吗？ :')

编辑

如果您想要一种更 OOP 的方法，您可以将上面的代码转换成类似这样的形式：

import random


class Goblin(object):

    def __init__(self):
        self.Name = "Goblin"
        self.HP = random.randint(15, 20)
        self.Damage = random.randint(5, 10)


class Army(object):

    def __init__(self):
        self.entities = []

    def remove(self, target):
        if target in self.entities:
            self.entities.remove(target)

    def filter(self, target):
        self.entities = filter(
            lambda entity: self.entities != entity, self.entities)


class BattleSpace(object):

    def __init__(self):
        self.armies = []

    def remove(self, target):
        for index, army in enumerate(self.armies):
            army.remove(target)

    def filter(self, target):
        for index, army in enumerate(self.armies):
            army.filter(target)


x = Goblin()
battle_space = BattleSpace()
Friendly_List = Army()
Enemy_List = Army()
battle_space.armies = [Friendly_List, Enemy_List]
Enemy_List.entities.append(x)

print "Initial setup"
print Enemy_List.entities
print Friendly_List.entities

print "After removing"
battle_space.remove(x)
print Enemy_List.entities
print Friendly_List.entities

通过这种方式，您可以轻松地将您的 BattleSpace 扩展到来自不同星球的数以万计的星际军队