嵌套列表和计数()

时间:2011-04-29 05:06:57

标签: python list

我想获得x出现在嵌套列表中的次数。

如果列表是:

list = [1, 2, 1, 1, 4]
list.count(1)
>>3

这没关系。但如果列表是:

list = [[1, 2, 3],[1, 1, 1]]

如何获得1次出现的次数?在这种情况下,4。

8 个答案:

答案 0 :(得分:33)

>>> L = [[1, 2, 3], [1, 1, 1]]
>>> sum(x.count(1) for x in L)
4

答案 1 :(得分:11)

这是另一种展平嵌套序列的方法。一旦序列变平,就可以轻松检查项目的数量。

def flatten(seq, container=None):
    if container is None:
        container = []

    for s in seq:
        try:
            iter(s)  # check if it's iterable
        except TypeError:
            container.append(s)
        else:
            flatten(s, container)

    return container


c = flatten([(1,2),(3,4),(5,[6,7,['a','b']]),['c','d',('e',['f','g','h'])]])
print(c)
print(c.count('g'))

d = flatten([[[1,(1,),((1,(1,))), [1,[1,[1,[1]]]], 1, [1, [1, (1,)]]]]])
print(d)
print(d.count(1))

以上代码打印:

[1, 2, 3, 4, 5, 6, 7, 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
1
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
12

答案 2 :(得分:9)

itertoolscollections模块只获得了您需要的内容(使用itertools.chain展平嵌套列表并使用collections.Counter计数

import itertools, collections

data = [[1,2,3],[1,1,1]]
counter = collections.Counter(itertools.chain(*data))
print counter[1]

使用递归展平函数代替itertools.chain来展平任意级别深度的嵌套列表

import operator, collections

def flatten(lst):
    return reduce(operator.iadd, (flatten(i) if isinstance(i, collections.Sequence) else [i] for i in lst))
已使用reduce代替operator.iadd代替sum,以便展平仅构建一次并就地更新

答案 3 :(得分:5)

试试这个:

reduce(lambda x,y: x+y,list,[]).count(1)

基本上,您从空列表[]开始,并将列表list的每个元素添加到其中。在这种情况下,元素本身就是列表,你会得到一个扁平的列表。

PS:刚刚在另一个问题上得到了类似的答案!

PPS:刚刚推出了这个解决方案!

答案 4 :(得分:4)

对于它来说:计算任意嵌套深度,处理元组,列表和参数:

hits = lambda num, *n: ((1 if e == num else 0)
    for a in n
        for e in (hits(num, *a) if isinstance(a, (tuple, list)) else (a,)))

lst = [[[1,(1,),((1,(1,))), [1,[1,[1,[1]]]], 1, [1, [1, (1,)]]]]]
print sum(hits(1, lst, 1, 1, 1))

15

答案 5 :(得分:3)

如果只有一个级别的嵌套,可以使用此列表comprenension进行展平:

>>> L = [[1,2,3],[1,1,1]]
>>> [ item for sublist in L for item in sublist ].count(1)
4
>>> 

答案 6 :(得分:0)

def nested_count(lst, x):
    return lst.count(x) + sum(
        nested_count(l,x) for l in lst if isinstance(l,list))

此函数返回所有包含的子列表中的出现次数和递归嵌套计数。

>>> data = [[1,2,3],[1,1,[1,1]]]
>>> print nested_count(data, 1)
5

答案 7 :(得分:0)

以下函数将通过将非列表添加到结果输出列表中并递归处理列表来展平任何深度的列表清单。(a)

def flatten(listOrItem, result = None):
    if result is None: result = []      # Ensure initial result empty.

    if type(listOrItem) != type([]):    # Handle non-list by appending.
        result.append(listOrItem)
    else:
        for item in listOrItem:         # Recursively handle each item in a list.
            flatten(item, result)
    return result                       # Return flattened container.

mylist = flatten([[1,2],[3,'a'],[5,[6,7,[8,9]]],[10,'a',[11,[12,13,14]]]])
print(f'Flat list is {mylist}, count of "a" is {mylist.count("a")}')

print(flatten(7))

一旦列表扁平化,在其上使用count就很简单。 该代码的输出为:

Flat list is [1, 2, 3, 'a', 5, 6, 7, 8, 9, 10, 'a', 11, 12, 13, 14], count of "a" is 2
[7]

请注意,如果您不传递实际列表,则会假设您想要一个列表,而该列表仅包含单个项目。


如果您不想构建一个扁平化的列表,则可以使用类似的方法来获取列表列表中任何项目的数量,例如:

def deepCount(listOrItem, searchFor):
    if type(listOrItem) != type([]): # Non-list, one only if equal.
        return 1 if listOrItem == searchFor else 0

    subCount = 0                     # List, recursively collect each count.
    for item in listOrItem:
        subCount += deepCount(item, searchFor)
    return subCount

deepList = [[1,2],[3,'a'],[5,[6,7,[8,9]]],[10,'a',[11,[12,13,14]]]]
print(f'Count of "a" is {deepCount(deepList, "a")}')
print(f'Count of 13  is {deepCount(deepList, 13)}')
print(f'Count of 99  is {deepCount(deepList, 99)}')

如预期的那样,其输出为:

Count of "a" is 2
Count of 13  is 1
Count of 99  is 0

(a)当然,要达到Python本身施加的限制,可以通过将其添加到代码顶部来增加限制:

import sys
sys.setrecursionlimit(1001) # I believe default is 1000.

我提到,只是在案例中,您就有一些深深嵌套的结构,但您实际上并不需要它。如果您要深层嵌套,那么您可能做错了:-)