vendredi 17 juillet 2020

Remove different substrings from list of strings

I have a list of strings with two different prefixes that I would like to remove.

 example_list=[
    '/test1/test2/test3/ABCD_1',
    '/test1/test2/test3/ABCD_2',
    '/test1/test2/test3/ABCD_3',
    '/test1/test4/test5/test6/ABCD_4',
    '/test1/test4/test5/test6/ABCD_5',
    '/test1/test4/test5/test6/ABCD_6',
    '/test1/test4/test5/test6/ABCD_7']

I would like the new list to look like:

  example_list=[
    'ABCD_1',
    'ABCD_2',
    'ABCD_3',
    'ABCD_4',
    'ABCD_5',
    'ABCD_6',
    'ABCD_7']

I was trying something like this, but keep running into errors.

for i in example_list:
    if i.startswith('/test1/test2/test3/'):
        i=i[19:]
    else:
        i=i[25:]

Aucun commentaire:

Enregistrer un commentaire