对对象列表进行排序,每个对象都包含一个列表

时间:2015-03-20 06:42:34

标签: python sorting

我正在编写杜松子酒纸牌游戏,用于处理套牌和牌组合。一组是3或4张具有相同等级的牌(即9c,9d,9h)。合并是一组相同级别的牌(即8c,9c,10c)。我想对melds和sets的组合列表进行排序。

卡片的表示如下:

class Card:
  def __init__(self, rank, suit):
    self.rank = rank
    self.suit = suit

卡片存储在CardGroups中,如下所示:

class CardGroup:
  def __init__(self, card_list):
    self.cards = []
    for c in card_list:
      self.cards.append(c)

以下是 2c,2h,2d,4h,5d,5s,5c,6c,7c,8c的示例:

这方面的融合与俱乐部有关:(5c,6c,7c)(6c,7c,8c)。这些集合(2c,2d,2h)(5c,5d,5s)

结果列表可能像这样组合(melds,然后设置):

unsorted_list = [ CardGroup([Card(5,'c'), Card(6,'c'), Card(7,'c')], 
                  CardGroup([Card(6,'c'), Card(7,'c'), Card(8,'c')],
                  CardGroup([Card(2,'c'), Card(2,'d'), Card(2,'h')], 
                  CardGroup([Card(5,'c'), Card(5,'d'), Card(5,'s')] ]

结果列表应该存在(如卡片比较):

sorted_list   = [ CardGroup([Card(2,'c'), Card(2,'d'), Card(2,'h')], 
                  CardGroup([Card(5,'c'), Card(5,'d'), Card(5,'s')],
                  CardGroup([Card(5,'c'), Card(6,'c'), Card(7,'c')], 
                  CardGroup([Card(6,'c'), Card(7,'c'), Card(8,'c')] ]

请注意,第二个和第三个CardGroup都以 5c 开头。重要的是要正确排序。如果我们有像 2c,3c,4c,5c 这样的4集,我们也隐含地拥有3组 2c,3c,4c 。在对这两个进行排序时,我们希望3组在4组之前。

我试过的事情:

unsorted_list.sort(key=lambda x: x.cards)
unsorted_list.sort(key=lambda x: x.cards[0])
unsorted_list.sort(key=lambda x: tuple(x.cards))

感谢您的帮助!

编辑:建议使用比较运算符作为答案。以下是此类解决方案的单元测试:

Class TestCard:
    def test___cmp__(self):
        card1 = Card(5, 'd')
        card2 = Card(6, 'c')
        card3 = Card(6, 's')

        self.assertLessEqual(card1.__cmp__(card2), -1)
        self.assertLessEqual(card1.__cmp__(card3), -1)
        self.assertEqual(card1.__cmp__(card1), 0)

        self.assertGreaterEqual(card2.__cmp__(card1), 1)
        self.assertLessEqual(card2.__cmp__(card3), -1)

        self.assertGreaterEqual(card3.__cmp__(card1), 1)
        self.assertGreaterEqual(card3.__cmp__(card2), 1)

1 个答案:

答案 0 :(得分:1)

我相信这可能有所帮助:

class Card:
  # ...
  def __cmp__(self, other):
    r = self.rank.__cmp__(other.rank)
    if not r:
      return self.suit.__cmp__(other.suit)
    return r

class CardGroup:
  # ...
  def __cmp__(self, other):
    return cmp(self.cards, other.cards)

现在您应该只能使用unsorted_list.sort()

相关问题