Question

我有一个包含子列表的列表，我想搜索每个子列表的第一个元素，然后打印找到该元素的子列表。我的代码是：

list = [[1,2,3,4,5],[6,7,8,9,10]]

for i in list:
    if (i[0])==6:
        print "Yes"

程序显示我找到了元素，但是如何打印找到它的列表？（[6,7,8,9,10]）

Answer 1

您只需要打印i即可。您已经在for循环中使用了它：

for i in list:
    if (i[0])==6:
        print "Yes"
        print i

输出：

Yes
[6, 7, 8, 9, 10]

Answer 2

list = [[1,2,3,4,5],[6,7,8,9,10]]

for i in list:
    if (i[0])==6:
        print "Yes"
        print i

Answer 3

您可以使用类似闭包的方法：

def find_list(list_of_list, item):
    def find_item(item):
        if item == list_[0]: 
            print('%i found in %r ' % (item, list_))
            return list_

    for list_ in list_of_list:
        found_list = find_item(item)
        if found_list:
            return found_list


>>> print(find_list([[1,2,3,4,5],[6,7,8,9,10]], 1))
1 found in [1, 2, 3, 4, 5] 
[1, 2, 3, 4, 5]

Answer 4

my_list = [[1,2,3,4,5],[6,7,8,9,10]]

print [sub for sub in my_list if sub[0] == 6]

打印找到所需号码的子列表

4 个答案: