如果子字符串匹配,我想删除列表中的所有元素

时间:2020-04-09 12:50:21

标签: python list dataframe

我有一个如图所示的列表,并且想要删除该列表中的所有其他元素,但前提是“无”或“未选择”并且具有以下两个词:

List = ['None, Pink, Green', 'Blue, Not Chosen', 'Red', 'Blue', 'Green, Red']

我想要的列表:

List = ['None, None', 'Not Chosen, Not Chosen', 'Red', 'Blue', 'Green, Red']

1 个答案:

答案 0 :(得分:1)

很难理解您的意思,但是此代码将产生所需的输出:

List = ['None, Pink, Green', 'Blue, Not Chosen', 'Red', 'Blue', 'Green, Red']

def make_replace(item):
    if ‘None’ in item:
        return ‘None, None’
    if ‘Not Chosen’ in item:
        return ‘Not Chosen, Not Chosen‘
    return item

List = [make_replace(item) for item in List]