Question

Is it possible to replace the following code with a list comprehension expression?

input = ['1\t2,3\t4,5', '61\t7,8\t9,0']

res = []
li = [i.split() for i in input]
for i in li:
    l = [i[0]]
    l = l + [e.split(',') for e in i[1:]]
    res.append(l)

The problem is that the first element in every sublist should be treated differently than the rest of the elements.

Answer 1

I have to say this isn’t really that Pythonic considering readability.

>>> l = ['1\t2,3\t4,5', '61\t7,8\t9,0']
>>> [[i[0]]+[e.split(',') for e in i[1:]] for i in [x.split() for x in l]]
[['1', ['2', '3'], ['4', '5']], ['61', ['7', '8'], ['9', '0']]]