如何在特定条件下以特定顺序将列表放入一个更大的列表中？ -Python 3.x

Question

现在，我有一个代码看起来类似于以下内容：

def get_newdeck(choice, P1, P2, P3):
    deck = []
    if choice == 1:
        deck = P2+P1+P3
        return deck
    elif choice == 2:
        deck = P1+P2+P3
        return deck
    else:
        deck = P1+P3+P2
        return deck

这是代码的一部分，而该代码也是魔术技巧的代码的一部分，魔术技巧从21的“ deck”中猜测用户的牌。 choice是为其输入的用户输入（整数为1、2或3）该卡以前在其中， P1, P2, P3是代表这三堆的列表； 有关这些堆的更多信息，请参见下文。
算法卡技巧如下：

1. 魔术师将21张打乱的扑克牌正面朝上分成三堆，从左到右逐行工作。 魔术师第一次执行此操作时，要求观众选择其中一张卡，并对其选择保密。

2. 魔术师要求听众说出他们的牌是在第一堆，第二堆还是第三堆中。

3.魔术师一堆一堆地收集卡片，确保将选定的堆放置在其他两个堆之间。

4. 将步骤1-3再重复2次。

5. 完成这些步骤后，所选卡片将“神奇地”位于21张卡片组的中间。

有问题的步骤是第三步，我正在尝试使其看起来不太重复。

有没有办法让我的代码（如上所示）使用循环工作？ 仅供参考，我正在使用Python-3.x

如果需要的话，这里是完整的代码-我知道它很杂乱，没有注释，但是我正在尝试从“混乱”部分开始进行修复。

import random

def get_cards():
    nums, suits=['Ace','2','3','4','5','6','7','8','9','10','Jack','Queen','King'], ['Hearts','Clubs','Diamonds','Spades']
    cards_dealt=[]
    while len(cards_dealt) < 22:
        card = random.choice(nums) + " of " + random.choice(suits)
        if card not in cards_dealt:
            cards_dealt.append(card)
        else:
            continue        
    return cards_dealt


def get_piles(deck):
    P1, P2, P3 = [], [], []
    for i in range(21):
        if i%3 == 0:
            P1.append(deck[i])
        elif i%3 == 1:
            P2.append(deck[i])
        else:
            P3.append(deck[i])
    return P1, P2, P3

#####################################
def get_newdeck(choice, P1, P2, P3):
    deck = []
    if choice == 1:
        deck = P2+P1+P3
        return deck
    elif choice == 2:
        deck = P1+P2+P3
        return deck
    else:
        deck = P1+P3+P2
        return deck
#####################################

def main():
    deck = get_cards()
    print("Choose a card and remember it")
    for x in range(3):
        P1, P2, P3 = get_piles(deck)
        piles = {'Pile 1':P1, 'Pile 2': P2, 'Pile 3':P3} 
        for numpile, pile in sorted(piles.items()):
            print()
            print(numpile,': ')
            for card in pile:
                print(card)
        print()
        choice = 0
        while choice > 3 or choice < 1:
            while True:
                try:
                    choice = int(input("Which pile is your card in? (1-3)? > "))
                    break
                except ValueError:
                    print("Must be a number")
        deck = get_newdeck(choice, P1, P2, P3)
    print()
    print('Your card is {}'.format(deck[10]))


main()

Answer 1

您可以执行以下操作：

def get_newdeck(choice, P1, P2, P3):
    chunks = [P1, P2, P3]
    idx = choice-1
    deck = chunks[(idx-1) % 3]+chunks[idx % 3]+chunks[(idx+1) % 3]
return deck

我采用idx=choice-1所以他们的选择是0、1或2，这将与chunks列表中的三张纸牌很好地匹配。 现在， chunks[idx]是他们选择的桩， chunks[idx-1]和chunks[idx+1]分别是他们选择的桩之前和之后的桩，因此可以通过将桩放置在其他两个桩之间来创建deck 。 % （模运算）只是确保如果我们超出0,1,2范围，我们会将结果包装回该范围。

如果您对他们的选择进行了想象，这很容易形象化。然后，桩就按照P1，P2，P3的顺序排列。

如果用户选择第一个桩，则idx将为0，而在此之前的桩将为-1。 模数3变为2，这是序列中的最后一个堆。

Answer 2

首先，我不认为您的代码在3种情况下看起来都是重复的。 但是为了交换想法，我的想法是：

由于您希望choice位置与中间元素交换，在本例中为索引1 （包含三个元素的列表，索引将为0, 1, 2 ），因此您可以执行

piles[choice - 1], piles[1] = piles[1], piles[choice - 1]

（当然，如果choice = 2 ，则什么都不会改变）

测试代码：

def get_newdeck(choice, P1, P2, P3):
    piles = [P1, P2, P3]
    piles[choice - 1], piles[1] = piles[1], piles[choice - 1]
    deck = []
    for pile in piles:
        deck.extend(pile)
    return deck


if __name__ == "__main__":
    choice = 1
    P1 = [1, 2]
    P2 = [3, 4]
    P3 = [5, 6]
    deck = get_newdeck(choice, P1, P2, P3)
    print(deck)

Answer 3

我认为您的代码没有任何严重的问题。 对于包含3个元素的列表，如果您已经在做，而不是编写不必要的混乱循环，那么可读性就很高。

我在玩你的代码，我只做了几件事，主要是外观：)

import random

def get_cards():
    nums, suits=['A','2','3','4','5','6','7','8','9','10','J','Q','K'], [u'\u2665',u'\u2663',u'\u2666',u'\u2660']
    cards_dealt=[]
    while len(cards_dealt) < 22:
        card = random.choice(nums) + random.choice(suits)
        if card not in cards_dealt:
            cards_dealt.append(card)
        else:
            continue        
    return cards_dealt


def get_piles(deck):
    P1 = [deck[i] for i in range(0,21,3)]
    P2 = [deck[i] for i in range(1,21,3)]
    P3 = [deck[i] for i in range(2,21,3)]
    return P1, P2, P3

#####################################
def get_newdeck(choice, P1, P2, P3):
    deck = P1+P3+P2
    if choice == 1:
        deck = P2+P1+P3
    elif choice == 2:
        deck = P1+P2+P3
    return deck
#####################################

def main():
    deck = get_cards()
    print("Choose a card and remember it")
    for x in range(3):
        P1, P2, P3 = get_piles(deck)
        piles = {'Pile 1':P1, 'Pile 2': P2, 'Pile 3':P3} 
        for numpile, pile in sorted(piles.items()):
            print(numpile+': ', end='\t')
            for card in pile:
                print(card, end='\t')
            print()
        print()
        choice = 0
        while choice > 3 or choice < 1:
            while True:
                try:
                    choice = int(input("Which pile is your card in? (1-3)? > "))
                    break
                except ValueError:
                    print("Must be a number")
        deck = get_newdeck(choice, P1, P2, P3)
    print()
    print('Your card is {}'.format(deck[10]))

main()