Regex replace in List Comprehension in Python

Question

Does anyone know how to use re.sub in Python with list comprehensions.

Im using the following,

>>> a = ["N","!","1","abc"]
>>> [(re.sub(r'(h|N|!|N|1)', r"'\033[91m'\g<1>'\033[0m'", 'x')) for x in a]
['x', 'x', 'x', 'x']

As you can see Im only getting x returned as the list elements.

Thanks,

I see you're doing for x in a, but you never actually use the x variable. Is that intentional? — Kevin
– Kevin, Commented Aug 8, 2014 at 13:44

falsetru · Accepted Answer · 2014-08-08 13:49:24Z

5

As Kevin commented, you didn't used x, but used the string literal 'x':

>>> [(re.sub(r'(h|N|!|N|1)', r"'\033[91m'\g<1>'\033[0m'", x)) for x in a]
["'\x1b[91m'N'\x1b[0m'", "'\x1b[91m'!'\x1b[0m'", "'\x1b[91m'1'\x1b[0m'", 'abc']

UPDATE

The regular expression can be expressed using character class ([....]) if the components are all single-character strings.

>>> [(re.sub(r'([hN!1])', r"'\033[91m'\g<1>'\033[0m'", x)) for x in a]
["'\x1b[91m'N'\x1b[0m'", "'\x1b[91m'!'\x1b[0m'", "'\x1b[91m'1'\x1b[0m'", 'abc']

edited Aug 8, 2014 at 13:49

answered Aug 8, 2014 at 13:44

falsetru

371k69 gold badges769 silver badges659 bronze badges

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

Collectives™ on Stack Overflow

Regex replace in List Comprehension in Python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related