首页 > 解决方案 > 在两个列表之间随机划分项目

问题描述

我必须通过使用类和继承从列表中获取项目并将它们随机附加到两个列表中,所以我不知道如何将它们随机附加到列表中

import random

players = ['a', 'b', 'c','d','e', 'f']

A = []
B = []
teams = []



class Human:
    def __init__(self,name):
        self.name = name



class Player(Human):

    def __init__(self, name, team):
        self.team = team
        Human.__init__(self, name)

for i in range(6):
    p = Player(random.choice(players))
    a = p.team
    print(p.name)
    players.remove(p.name)
print(teams)

标签: python

解决方案


您的代码将为您提供相同的玩家:

for i in range(6):
    p = Player(random.choice(players)) # you can draw duplicates here

避免使用random.sample(players, k=len(players))您的全名(这将返回一个新的、打乱的列表 -players未修改)或random.shuffle(players)就地使用。之后,您将结果分为两部分并根据该数据创建团队:

您可能应该重载__repr__(self)类的方法以使输出更整洁。

import random

class Human:
    def __init__(self,name):
        self.name = name
    def __str__(self):
        return f"{self.name}"
    def __repr__(self): return str(self)

class Player(Human):
    def __init__(self, name, team):
        super().__init__(name)
        self.team = team
    def __str__(self):
        return f"{self.name} in team {self.team}" 
    def __repr__(self): return str(self)

players = ['a', 'b', 'c','d','e', 'f']
random.shuffle(players)  # inplace shuffling

player_length = len(players) // 2 # make sure you got even player counts - else uneven teams
A = [Player(k,"A") for k in players[:player_length]]  # first 3 are Team A
B = [Player(k,"B") for k in players[player_length:]]  # rest is Team B
teams = [A,B]

print(teams)

输出:

[[d in team A, f in team A, b in team A],[a in team B, e in team B, c in team B]]

独库:


推荐阅读