我正在使用列表列表(类似于拼图)并尝试编写一个函数,如果在与另一个列表(可以看作列)相同的索引中至少有一个重复值,该函数将返回“False” ,视觉上)(否则为真)。例如:
List = [[ 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10],
[11, 12, 13, 14, 15],
[16, 17, 18, 19, 20],
[21, 22, 23, 24, 25]]
应该返回 'True' 因为没有重复 但是:
List = [[ 1, 2, 3, 4, 5],
[ 6, 2, 8, 9, 10],
[11, 12, 13, 14, 15],
[16, 17, 18, 19, 20],
[21, 22, 23, 24, 25]]
应该返回“False”,因为在 index[1] 的 list1 中有值“2”,在索引为 [1] 的 list2 中也有值“2”。
下面是我写函数的方式
def check_columns_valid(list):
for i in range(len(list)):
if i[0] == i[5] == i[10] == i[15] == i[20]:
return False
elif i[1] == i[6] == i[11] == i[16] == i[21]:
return False
elif i[2] == i[7] == i[12] == i[17] == i[22]:
return False
elif i[3] == i[8] == i[13] == i[18] == i[23]:
return False
elif i[4] == i[9] == i[14] == i[19] == i[24]:
return False
return True
我试图检查某些索引是否彼此相等,但如果所有索引都彼此相等并且我只需要至少一个,我的代码可能只会返回“False”。而且我也不确定这样做是否有效。我不知道我是否在跳过步骤。
请您参考如下方法:
您可以使用 zip 和 sets 来帮助解决这个问题。
def check_columns_valid(puzzle):
# Iterate over the puzzle one column at a time.
for column in zip(*puzzle):
# Get all of the unique values in the column.
uniques = set(column)
if len(uniques) != len(column):
# If the number of unique values isn't the same as the
# number of values, return False.
return False
return True
或者,如果您是单线的粉丝,所有这些都可以简化为:
def check_columns_valid(puzzle):
return all(len(x) == len(set(x)) for x in zip(*puzzle))
根据拼图的大小,如果您使用的是 Python 2.x,您可能需要考虑使用 izip(from itertools import izip) 而不是 zip。
