Будучи в состоянии обнаружить int в строках, Python

Question

Я пытаюсь отделить число от строки в Python. В принципе, я хочу, чтобы, если color == 'grey' + число, то оно вернет это число. Например, если цвет равен 'grey23', он вернет 23. Если цвет равен 'серый', это вызовет оператор else.

псевдокод:

# = an int

def func (color):
    if color == 'gray' and a # :
        return int(#)
    else:
        print 'pass'

Answer 1

import re

def func (color):
    try:
        return int(re.search('(\d+)$',color).group(0)))
    except AttributeError:
        print 'pass'

Answer 2

Вы можете использовать регулярные выражения для этого:

import re

matches = re.match('\w+(\d+)', color)
result = matches.groups()
if(len(result) > 0):
    return result[0]
else:
    return 'pass'

Не проверено, поэтому может содержать ошибки, но это основная суть.

Answer 3

if color.startswith("grey") and color[4:].isdigit():
    return int(color[4:])
else:
    return 'pass'

color[4:] может быть заменено чем-то более общим, чем жестко закодированное значение 4, но, поскольку "серый" (или "серый" - вы используете оба) жестко закодирован, казалось, с этим проблем нет.