Breakdown nested for loops in Python list comprehension that involves string concatenation

Question

I am trying to debug the nested for loop via Pycharm debugger... In the process of troubleshooting, I wanted to breakdown the loop into two individual loops and step through the code but I am having a hard time with this...

Here is the code block with list comprehension:

def letterCasePermutation(S):
    res = ['']
    for ch in S:
        if ch.isalpha():
            res = [i + j for i in res for j in [ch.upper(), ch.lower()]]
    return res

result = letterCasePermutation("ab")
print(result) # expected result = ['AB', 'Ab', 'aB', 'ab']

In order to debug this code block I would like to break down the list comprehension to something like this:

def letterCasePermutation(S):
    res = ['']
    for ch in S:
        if ch.isalpha():
            # res = [i + j for i in res for j in [ch.upper(), ch.lower()]]

            for i in res:
                for j in [ch.upper(), ch.lower()]:
                    res.append(i + j)
    return res

result = letterCasePermutation("ab")
print(result)

The above block results in an infinite loop error instead of providing the result like code block-1. expected result = ['AB', 'Ab', 'aB', 'ab']

I am not able to figure what I am missing. After spending considerable amount of time and still being stuck, I decided to post this question. Thanks for the help in advance.

Siddhant · Accepted Answer · 2019-10-16 02:27:37Z

1

It results in an infinite loop because you are iterating on res, for i in res, and appending new values in it at the same time res.append(i + j).

Which is not the case with list-comprehension since expression on the right of = is evaluated and assigned to res.

You can use a second list to avoid doing that like so,

def letterCasePermutation(S):
res = ['']
for ch in S:
    if ch.isalpha():
        _res = []
        for i in res:
            for j in [ch.upper(), ch.lower()]:
                _res.append(i + j)
        res = res + _res
return res

result = letterCasePermutation("ab")
print(result)

Edit:

def letterCasePermutation(S):
res = ['']
for ch in S:
    if ch.isalpha():
        _res = []
        for i in res:
            for j in [ch.upper(), ch.lower()]:
                _res.append(i + j)
        res = _res
return res
result = letterCasePermutation("ab")
print(result)

edited Oct 16, 2019 at 2:27

answered Oct 15, 2019 at 3:26

Siddhant

5936 silver badges14 bronze badges

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

2 Comments

InquisitiveGirl Over a year ago

Thank you @Siddhant for taking the time to answer the question. What still puzzles me is while returning the result how can I dynamically splice ? I tried playing around with the above code you posted and I get the result as ['', 'A', 'a', 'B', 'b', 'AB', 'Ab', 'aB', 'ab'] which is incorrect

InquisitiveGirl Over a year ago

Yep i got it now

Chillie · Accepted Answer · 2019-10-15 03:28:07Z

Comprehensions don't care about assinging to the same name as used in the comprehension.

a = [0, 1, 2, 3, 4]
a = [i*2 for i in a]
print(a)

Outputs [0, 2, 4, 6, 8].

With your example you are adding elements to the res list while iterating over it:

for i in a:
    a.append(i)

This gives you an infinite loop because as you proceed to the next element, more elements are added to the list.

Your options are either assigning to a new variable name, or using slicing to iterate over a temporary copy of the list:

a = [0, 1, 2, 3, 4]
b = []
for i in a:
    b.append(i)

print(b)

Outputs [0, 1, 2, 3, 4]

a = [0, 1, 2, 3, 4]
for i in a[:]:
    a.append(i)

print(a)

Output is [0, 1, 2, 3, 4, 0, 1, 2, 3, 4].

a[:] is a slice of a from the first to the last element with a step of 1. You can read more about slicing here or in the official python docs.

InquisitiveGirl · Accepted Answer · 2019-10-16 02:25:30Z

0

Here is the breakdown :) thank you both for helping me with the idea.

def letterCasePermutation(S):
    res = ['']
    for ch in S:
        _res = []
        for i in res:
            for j in [ch.upper(), ch.lower()]:
                _res.append(i + j)
        res = _res
    return res


result = letterCasePermutation("ab")
print(result)

answered Oct 16, 2019 at 2:25

InquisitiveGirl

6974 gold badges13 silver badges33 bronze badges

Collectives™ on Stack Overflow

Breakdown nested for loops in Python list comprehension that involves string concatenation

3 Answers 3

2 Comments

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

2 Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related