Extract text with multiple separators

Question

I have list of strings with separators A and B:

L = ['sgfgfqds A aaa','sderas B ffff','eeee','sdsdfd A rrr']

and need:

L1 = [['aaa'], ['ffff'], ['eeee'], ['rrr']] 

I tried using:

L1 = [re.findall(r'(?<=A)(.*)$', inputtext) for inputtext in L]
print (L1)

but, it returns the following:

[[' aaa'], [], [], [' rrr']] 

How can I get the desired output?

Answer 1

You can use re.split to break your strings on either 'A' or 'B':

>>> L1 = [re.split(r'[AB] *', inputtext)[-1] for inputtext in L]
>>> L1
['aaa', 'ffff', 'eeee', 'rrr']

Answer 2

You can use the fact that split returns a list even if it doesn't find the separator.

 L1 = [[x.split(' A ')[-1].split(' B ')[-1]] for x in L]