2

I'm trying to flatten a tuple with mixed types into a list. The following function does not produce the desired output:

a = (1, 2, 3, ['first', 'second'])
def flatten(l): 
return flatten(l[0]) + (flatten(l[1:]) if len(l) > 1 else []) if type(l) is list else [l]

>>> flatten(a)
[(1, 2, 3, ['first', 'second'])]
>>> flatten(flatten(a))
[(1, 2, 3, ['first', 'second'])]
>>> [flatten(item) for item in a]
[[1], [2], [3], ['first', 'second']]

the output should be:

>>> [1, 2, 3, 'first', 'second']
Bob
  • 849
  • 5
  • 14
  • 26

1 Answers1

0
def flatten(l):
    if isinstance(l, (list,tuple)):
        if len(l) > 1:
            return [l[0]] + flatten(l[1:])
        else:
            return l[0]
    else:
        return [l]

a = (1, 2, 3, ['first', 'second'])

print(flatten(a))

[1, 2, 3, 'first', 'second']

Rahn
  • 4,787
  • 4
  • 31
  • 57