如何让我的代码将输入作为一组元组，并且仍然返回相同的答案？

Question

所以我试着写两段代码，其中一段试图找出一个代码序列是否是fourz。

Fourz基本上可以看到一张卡片是否有4个相同的第一个字符和一个不相同的字符。

例如，

fourz(['3S', '3D', '3C', '3H', '5S']) returns True because there are 4 3 and one 5
fourz(['4S', '3D', '3C', '3H', '5S']) returns False because there are only three 3, and one 4 and one 5.

现在我有了这个代码：

def is_four_of_a_kind(h):
    h = list(h)
    values = [i[0] for i in h]
    card_order_dict = {'2': 2, '3':3, '4':4, '5': 5, '6':6, '7':7,
                       '8':8, '9':9,'T':10, 'J':11, 'Q':12, 'K':13, 'A':14 }
    count = 1
    rank_values = [card_order_dict[i] for i in values]
    rank_values = sorted(rank_values)

    for i in range(1,len(rank_values)):
        if rank_values[i] == rank_values[0]:
            count += 1
    if count == 4:
        return True
    else:
        return False

当代码是一个列表时，此代码正确返回输出，但当它是一组元组时，此代码不起作用

所以如果我的代码是这样的fourz（{（'3S'，'3D'，'3C'，'3H'，'5S'）}。。

我的代码需要做哪些更改才能正常工作？？

Answer 1

这是因为您正在集合中定义一个元组。因此，当您将列表转换为列表时，列表中只有一个元素。

为了使代码工作，您可以对转换到列表的功能进行简单的更改。

def is_four_of_a_kind(h):
if type(h) == set:
    for i in h:
        h = list(i)
else:
    h = list(h)

values = [i[0] for i in h]
card_order_dict = {'2': 2, '3':3, '4':4, '5': 5, '6':6, '7':7,
                   '8':8, '9':9,'T':10, 'J':11, 'Q':12, 'K':13, 'A':14 }
count = 1
rank_values = [card_order_dict[i] for i in values]
rank_values = sorted(rank_values)

for i in range(1,len(rank_values)):
    if rank_values[i] == rank_values[0]:
        count += 1
if count == 4:
    return True
else:
    return False