是否可以在python中获取列表中的所有元素?

2024-10-03 13:24:30 发布

您现在位置:Python中文网/ 问答频道 /正文

是否有一个函数根据搜索返回列表中的数组?你知道吗

例如,我想得到包含字母A的表的列表

myLst = [[ABC], [BCV], [BAD]]
return [[ABC], [BAD]]

我必须做我自己的功能吗?你知道吗


Tags: 函数功能列表return字母数组badabc
2条回答

很可能,也很简单,只需按以下步骤操作

if x for x in list if 'a' in x:
     #Do something

这是一个简单的列表理解,我建议在开始编写代码之前阅读Python MDN

您可以在一行中完成:

print([item for item in myLst for sub_item in item if 'A' in sub_item])

输出:

[['ABC'], ['BAD']]

或者如你所说你想要一个函数,下面是详细的解决方案:

def return_list(list_1):
    result=[]
    for item in list_1:
        if isinstance(item,list):
            for sub_item in item:
                if 'A' in sub_item:
                    result.append(item)
    return result

print(return_list(myLst))

相关问题 更多 >