0

假设我有一个列表，其中每个列表元素由三个部分组成，例如：

[[0, 0, 1], [0, 1, 0], [1, 0, 0], [0, 0, 2], [0, 1, 1], [0, 2, 0], [1, 0, 1], [1, 1, 0], [2, 0, 0], [0, 0, 3], [0, 1, 2], [0, 2, 1], [0, 3, 0], [1, 0, 2], [1, 1, 1], [1, 2, 0], [2, 0, 1], [2, 1, 0], [3, 0, 0], [0, 0, 4], [0, 1, 3], [0, 2, 2], [0, 3, 1], [0, 4, 0], [1, 0, 3], [1, 1, 2], [1, 2, 1], [1, 3, 0], [2, 0, 2], [2, 1, 1], [2, 2, 0], [3, 0, 1], [3, 1, 0], [4, 0, 0]]

有没有办法检查每个列表元素内部的内容，即说我想创建一个新列表，基于上面包含所有包含两个零的元素的索引位置，以及包含一个零的每个元素的列表，我该怎么做？

我知道如何检查单个事物是否在列表元素中，但不知道该元素是否出现两次。

4

4 回答 4

3

您可以使用该.count方法。

two_zeros = [x for x in lst if x.count(0) == 2]
one_zero = [x for x in lst if x.count(0) == 1]

如果您想变得非常聪明，您可以使用 collections.defaultdict 在一个循环中完成所有操作：

d = collections.defaultdict(list)
for sublist in lst:
    d[sublist.count(0)].append(sublist)

现在你有一个零数量到包含该数量零的子列表的映射。

当然，如果你真的想要一个索引列表，你可以使用enumerate.

于 2013-06-18T14:50:59.107 回答
2

使用count().

来自help：

count(...)
    L.count(value) -> integer -- return number of occurrences of value

示例代码（列表的一部分） -

>>> startList = [[0, 0, 1], [0, 1, 0], [1, 0, 0], [0, 0, 2], [0, 1, 1], [0, 2, 0], [1, 0, 1], [1, 1, 0], [2, 0, 0], [0, 0, 3], [0, 1, 2], [0, 2, 1]]
>>> for element in startList:
        element.count(0)
2
2
2
2
1
2
1
1
2
2
1
1

如何创建您的列表？将上述想法与列表理解一起使用。

>>> twoZero = [index for index, elem in enumerate(startList) if elem.count(0) == 2]
>>> twoZero
[0, 1, 2, 3, 5, 8, 9]
>>> oneZero = [index for index, elem in enumerate(startList) if elem.count(0) == 1]
>>> oneZero
[4, 6, 7, 10, 11]

这是您列表的一部分。

于 2013-06-18T14:50:50.077 回答
0

您可以使用列表理解、list.count和enumerate：

two_zeros_index = [i for i, item in enumerate(datalist) 
                   if item.count(0) == 2]

但是因为你想创建两个列表，一个普通的 for 循环可能会更好（所以你只迭代datalist一次）：

one_zero, two_zeros_index = [], []
for i, item in enumerate(datalist): 
    n = item.count(0)
    if n == 1:
        # Every *item* containing one zero
        one_zero.append(item)
    elif n == 2: 
        # Every *index* containing two zeros
        two_zeros_index.append(i)
于 2013-06-18T14:52:02.020 回答
0

该列表reqd将包含您需要的所有索引：

>>> reqd = []
>>> for i in range(len(d)):

    if d[i].count(0) == 2:
        reqd.append(i)


>>> reqd
[0, 1, 2, 3, 5, 8, 9, 12, 18, 19, 23, 33]
于 2013-06-18T14:52:48.143 回答