Get only numbers from strings that are in the list

Question

So I have list that looks like this

mylist = ['storeitem_1','storeitem_2','storeitem_3']

How would I go about getting only numbers from those strings? I tried following but it came out totally wrong.

for items in mylist:
  print(items.split("storeitem_"))

Any help is appreciated

Answer 1

Here is one way with a regex:

import re

mylist = ['storeitem_1','storeitem_2','storeitem_3']

re.findall('\d+', '|'.join(mylist))

Which gives:

['1', '2', '3']

Alternatively, something more along the lines of what you were trying:

[i.split('_')[-1] for i in mylist]

Gives you the same:

['1', '2', '3']

Answer 2

Use regular expressions:

import re
num_list = [re.search(r'\d+', x)[0] for x in mylist]

Answer 3

Assuming mylist always contains strings in the form of 'storeitem_n' where n is an integer from 0 to 9

You can do this to get n:

mylist = ['storeitem_1','storeitem_2','storeitem_3']

for items in mylist:
  print(items[-1])