如何维护一套?

时间:2012-07-14 02:05:05

标签: python set

我见过the related question,但我不能用frozenset作为我的内集。我希望一切都变得可变。

据我了解,Python提供的所有内容都为ID,为什么不能使用as its hash?我需要维护一个对所有内部集的引用列表,即使它们发生了变化。

编辑:好的,我理解为什么,但在这种情况下,它会更好,因为我只关心引用相等,而不是值相等。

我该怎么做?


您是  要问“为什么”,所以我会给你一些代码:

def remove_captured(self):
    all_chains = set()
    chains = Grid(self.rows, self.cols)
    for m, n, stone in self.enumerate():
        if stone == self[m - 1, n]:
            chains[m, n] = chains[m - 1, n]
            if stone == self[m, n - 1]:
                all_chains.discard(chains[m, n - 1])
                chains[m, n].update(chains[m, n - 1])
                for s in chains[m, n - 1]:
                    chains[s] = chains[m, n]
        elif stone == self[m, n - 1]:
            chains[m, n] = chains[m, n - 1]
        else:
            chains[m, n] = set()
            all_chains.add(chains[m, n])
        chains[m, n].add((m,n))
    chains._print()
    print all_chains

我基本上有一个游戏板,我想把棋盘上的棋子分成小组(或“链子”)。上面的代码工作正常,直到我添加all_chains - 它创建了所有集合,但是我无法访问它创建的每个集合,而无需再次遍历整个板。

那么如何维护它创建的所有集合的列表?请记住,我也需要删除集合(这就是为什么我要使用另一个集合作为外集)。


weakref.ref()中包含引用也不起作用:

all_chains.add(weakref.ref(chains[m, n])) # TypeError: unhashable type: 'set'

1 个答案:

答案 0 :(得分:1)

决定使用集合词典。

def remove_captured(self):
    cdict = {}
    cid = 0
    chains = Grid(self.rows, self.cols)
    for m, n, stone in self.enumerate():
        if stone == self[m - 1, n]:
            chains[m, n] = chains[m - 1, n]
            if stone == self[m, n - 1] and chains[m, n] != chains[m, n - 1]:
                del_id = chains[m, n - 1]
                cdict[chains[m, n]].update(cdict[del_id])
                for c in cdict[del_id]:
                    chains[c] = chains[m, n]
                del cdict[del_id]
        elif stone == self[m, n - 1]:
            chains[m, n] = chains[m, n - 1]
        else:
            cdict[cid] = set()
            chains[m, n] = cid
            cid += 1
        cdict[chains[m, n]].add((m, n))
    chains._print()
    pprint(cdict)

它不是那么漂亮,因为在我可以使用它之前我总是要在字典中查找它,但它似乎有用。

输入:

0  .  W  W  W  W
1  W  B  B  B  .
2  .  .  .  W  .
3  .  B  B  W  W
4  .  .  B  W  .
   0  1  2  3  4

输出:

0 0 1 1 1 1
1 2 3 3 3 4
2 5 5 5 6 4
3 5 7 7 6 6
4 5 5 7 6 8
  0 1 2 3 4
{0: set([(0, 0)]),
 1: set([(0, 1), (0, 2), (0, 3), (0, 4)]),
 2: set([(1, 0)]),
 3: set([(1, 1), (1, 2), (1, 3)]),
 4: set([(1, 4), (2, 4)]),
 5: set([(2, 0), (2, 1), (2, 2), (3, 0), (4, 0), (4, 1)]),
 6: set([(2, 3), (3, 3), (3, 4), (4, 3)]),
 7: set([(3, 1), (3, 2), (4, 2)]),
 8: set([(4, 4)])}