Python String and variable format validation

Question

I have a list:

data = ["data 01", "data 02", "data 03"]

I want to check if each element in list contain the string "data", and also add the text after data (e.g 01, 02, 03) to another list. Below is the code I have, but I find it to be inefficient.

data_2 = []

for item in data:
    if ("data" in item):
        data_2.append(item.replace("data", "").strip())

Is there a way to check

if item == "data {}"

Where {} is a numerical variable. As a precaution to having an item in data list like ["no data"], where the string "no" gets added by mistake?

There are numerous ways, but I would say that Python re is probably the way to go. — Johan Schiff
– Johan Schiff, Commented Jun 17, 2020 at 21:12

Red · Accepted Answer · 2020-06-17 21:20:04Z

1

You can use a list comprehension:

data = ["data 01", "data 02", "data 03"]
d = [s.split('data ',1)[1] for s in data if 'data' in s and s.split('data ',1)[1].isdigit()]
print(d)

Output:

['01', '02', '03']

edited Jun 17, 2020 at 21:20

answered Jun 17, 2020 at 21:09

Red

27.7k8 gold badges44 silver badges63 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Barmar Over a year ago

This still doesn't check that there's a number after data. Did you miss that part of the question?

Barmar · Accepted Answer · 2020-06-17 21:13:49Z

0

use a regular expression to test that it matches the whole pattern.

import re

for item in data:
    result = re.match(r'data (\d+)$')
    if result:
        data_2.append(result.group(1))

result.group(1) will contain the number that was captured by (\d+)

answered Jun 17, 2020 at 21:13

Barmar

789k57 gold badges555 silver badges669 bronze badges

Collectives™ on Stack Overflow

Python String and variable format validation

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related