很多时候,我们需要计算列表中存在的元素以进行某些数据处理。但是可能存在嵌套列表的情况,并且计数可能不直接。在本文中,我们将看到如何处理计算列表中元素数量的这些复杂性。
在这种方法中,我们使用两个for循环来遍历list的嵌套结构。在下面的程序中,我们有嵌套列表,其中内部元素内部包含不同数量的元素。我们还应用该len()
函数来计算平展列表的长度。
listA = [[2,9, 6], [5, 'a'], [0], [12,'c', 9, 3]] # Given list print("Given list : ",listA) res = len([x for y in listA for x in y]) # print result print("Total count of elements : " ,res)
输出结果
运行上面的代码给我们以下结果-
Given list : [[2, 9, 6], [5, 'a'], [0], [12, 'c', 9, 3]] Total count of elements : 10
在这种方法中,我们应用链方法,通过展平所有内部元素,然后将其转换为列表,从而从列表中取出所有内部元素。最后应用该len()
函数,以便找到列表中元素的数量。
from itertools import chain listA = [[2,9, 6], [5, 'a'], [0], [12,'c', 9, 3]] # Given list print("Given list : ",listA) res = len(list(chain(*listA))) # print result print("Total count of elements : " ,res)
输出结果
运行上面的代码给我们以下结果-
Given list : [[2, 9, 6], [5, 'a'], [0], [12, 'c', 9, 3]] Total count of elements : 10