的Python:测试在列表嵌套某些指标的平等
问题描述:
我有以下嵌套列表:的Python:测试在列表嵌套某些指标的平等
nlist = [[2, 0, 4], [2, 0, 4], [0, 0, 4], [0, 0, 4]]
我想评估每个列表的第一个元素是否相等。我找到了评估整个嵌套列表是否相等但不是特定索引的方法?
答
我会使用all()
函数与generator expression作为其参数。如果每个sublist[0] == nlist[0][0]
all(sublist[0] == nlist[0][0] for sublist in nlist)
总之,all()
将返回True
在原始列表中的每个子表。如果有任何项目不等于nlist[0][0]
,它将返回False
。
+0
我不熟悉all()函数。这工作完美。将在十分钟内接受。谢谢。 – spiral01
答
你可以做一个集中的每个子列表的第一要素,而由于在套不重复的,你可以看到,如果集的长度是:
nlist = [[2, 0, 4], [2, 0, 4], [0, 0, 4], [0, 0, 4]]
result = len(set(l[0] for l in nlist)) == 1
print(result) # => False
答
,你可以找到所有独特的第一个元素,然后检查指标:
>>> from itertools import groupby
>>> nlist = [[2, 0, 4], [2, 0, 4], [0, 0, 4], [0, 0, 4]]
>>> c= [i for i,k in groupby([n[0] for n in nlist])]
>>> c
[2, 0]
>>> d=[[i for i, x in enumerate(nlist) if x[0]==j] for j in c]
>>> d
[[0, 1], [2, 3]]
>>> R=dict(zip(c, d))
>>> R
{0: [2, 3], 2: [0, 1]}
你得到每个第一可能值和指数从最初的清单列表的dict
。
答
first_num = nlist[0][0]
for sublist in nlist:
if first_num != sublist[0]
print(False)
print(True)
'等于'什么? – Ajax1234