如何在列表列表中获取相同值的所有出现的索引列表？

Question

有一个列表，

mylist = [[1, 3, 4], [3, 6, 7], [8, 0, -1, 3]]

我需要一个方法来获取包含某些特定值的所有元素的索引的列表，例如“3”。 所以对于'3'的值，这样的索引列表应该是

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

提前致谢。

更新 ：我们可以在同一个子列表中有几个寻求值的实例（在我们的例子中为'3'），例如

my_list = [[1, 3, 4], [3, 6, 7], [8, 0, -1, 3, 3]]

所以期望的输出将是[[0, 1], [1, 0], [2, 3], [2, 4]] 。

我想我的解决方案有点天真，这里是：

my_list = [[1, 3, 4], [3, 6, 7], [8, 0, -1, 3, 3]]
value = 3
list_of_indexes = []

for i in range(len(my_list)):
        for j in range(len(my_list[i])):
            if my_list[i][j] == value:
                index = i, j
                list_of_indexes.append(index)

print list_of_indexes

>>[(0, 1), (1, 0), (2, 3), (2, 4)]]

看到更紧凑的解决方案会很棒

Answer 1

假设值在每个子列表中出现一次，您可以使用以下函数，该函数使用内置enumerate函数和列表推导：

my_list = [[1, 3, 4], [3, 6, 7], [8, 0, -1, 3]]

def func(my_list, x):
    return [[idx, sublist.index(x)] for idx, sublist in enumerate(my_list)]

answer = func(my_list, 3)
print(answer)

产量

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

Answer 2

简单的方法，

def get_val(mylist, val):
    for ix, item in enumerate(mylist):
        yield [ix, item.index(val)]

mylist = [[1, 3, 4], [3, 6, 7], [8, 0, -1, 3]]

val = list(get_val(mylist, 3))
print(val)

输出：

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