计算嵌套列表中项目的数量
我有一个列表,这个列表里面又包含了其他列表,下面是一个例子:
h = [[('red', array([ 1., 1., 1., ..., 0., 0., 0.])),
('blue', array([ 0., 0., 0., ..., 0., 0., 0.]))],
[('red', array([ 0., 0., 0., ..., 0., 0., 0.]))],
[('red', array([ 0., 0., 0., ..., 0., 0., 0.]))]]
我该如何计算每个子列表中,根据标签来统计项目的数量呢?比如说,在 h[1]
中,包含 'red'
的项目数量是10?上面的例子中有两个子列表,一个包含三个元素,另一个包含一个元素。
这样在 h[0]
中统计包含 'red' 的项目数量会得到2,而在 h[1]
中则是1。
1 个回答
1
如果你的数据结构是这样的:
goat = [('a', np.array([1,2,3])), ('b', np.array([1,2]))]
那么你可以得到:
[(i, len(list(j))) for i, j in goat]
输出结果将会是:
[('a', 3), ('b', 2)]