使用Python分离列表数据

时间:2019-07-03 06:49:50

标签: python python-3.x list loops

我有一个包含多个字符串的列表),我想用:分隔它们:

MainList:

[
    GENERAL NOTES & MISCELLANEOUS DETAILS_None_None_None, 
    STR_XX_XX_0001, 
    STR_XX_XX_0002, 
    STR_XX_XX_0003, 
    GENERAL ARRANGEMENT_None_None_None, 
    STR_XX_XX_10001.0, 
    STR_XX_XX_10002.0, 
    STR_XX_XX_10003.0, 
    STR_XX_XX_10004.0, 
    STR_XX_XX_10005.0, 
    STR_XX_XX_10006.0

]

如果在主列表中找到了字符串"_None_None_None",则可以将此数据添加到新的空列表中,并将剩余的STR_XX_XX_0001值添加到另一个列表中,直到找到另一个具有“ _None_None_None”的字符串为止,然后执行一样。

我已经尝试过自己,但是当它用"_None_None_None"找到下一个字符串时,我想我将无法中断循环。只是想出办法,不确定逻辑是否正确。

    empty1 = []
    empty2 = []

    for i in MainList:
            if "_None_None_None" in i:
               empty1.append(i)
            # Need help on hear onwards
            else:
                while "_None" not in i:
                    empty2.append(i)
                break

我希望输出结果在两个列表中。像这样:

列表1:

[
    GENERAL NOTES & MISCELLANEOUS DETAILS_None_None_None, 
    GENERAL ARRANGEMENT_None_None_None
]

列表2:

[
[STR_XX_XX_0001,STR_XX_XX_0002,STR_XX_XX_0003],[STR_XX_XX_10001.0,STR_XX_XX_10002.0,STR_XX_XX_10003.0,STR_XX_XX_10004.0,STR_XX_XX_10005.0,STR_XX_XX_10006.0]
]

List2是带有子列表的列表

2 个答案:

答案 0 :(得分:3)

这使它有些复杂,您可以让列表贯穿整个过程,而无需内部while循环。只需为循环中显示的每个元素做出决定:

empty1 = []
empty2 = []

for i in MainList:
    if "_None_None_None" in i:
        empty1.append(i)
    else:
        empty2.append(i)

这将为您提供两个列表:

> empty1
> ['GENERAL NOTES & MISCELLANEOUS DETAILS_None_None_None',
 'GENERAL ARRANGEMENT_None_None_None']

> empty2
> ['STR_XX_XX_0001',
 'STR_XX_XX_0002',
 'STR_XX_XX_0003',
 'STR_XX_XX_10001.0',
 'STR_XX_XX_10002.0',
 'STR_XX_XX_10003.0',
 'STR_XX_XX_10004.0',
 'STR_XX_XX_10005.0',
 'STR_XX_XX_10006.0']

根据评论进行编辑

如果注释者正确,并且您要将非NONE值分组到单独的列表中,那么这是itertools.groupby的一个好用例。它将以方便,有效的方式为您创建组,并且您的循环看起来几乎相同:

from itertools import groupby

empty1 = []
empty2 = []

for k, i in groupby(MainList, key = lambda x: "_None_None_None" in x):
    if k:
        empty1.extend(i)
    else:
        empty2.append(list(i))

这将为您提供相同的empty1,但empty2不会是列表列表:

[['STR_XX_XX_0001', 'STR_XX_XX_0002', 'STR_XX_XX_0003'],
 ['STR_XX_XX_10001.0',
  'STR_XX_XX_10002.0',
  'STR_XX_XX_10003.0',
  'STR_XX_XX_10004.0',
  'STR_XX_XX_10005.0',
  'STR_XX_XX_10006.0']]

答案 1 :(得分:1)

您可以尝试以下代码段:

dlist = ["GENERAL NOTES & MISCELLANEOUS DETAILS_None_None_None","STR_XX_XX_0001","STR_XX_XX_0002","STR_XX_XX_0003", "GENERAL ARRANGEMENT_None_None_None","STR_XX_XX_10001.0","STR_XX_XX_10002.0", "STR_XX_XX_10003.0", "STR_XX_XX_10004.0", "STR_XX_XX_10005.0", "STR_XX_XX_10006.0"]

with_None = [elem for elem in dlist if elem.endswith("_None")]
without_None = [elem for elem in dlist if not elem.endswith("_None")]

您还可以为该过程编写通用函数:

def cust_sept(src_list, value_to_find,f):
    with_value, without_value = [elem for elem in dlist if f(elem,value_to_find)],[elem for elem in dlist if not f(elem,value_to_find)]
    return with_value,without_value

list_one,list_two = cust_sept(dlist,"_None",str.endswith)