Let's say I have the following list data and I want to make the list flat_data:
data = [[1, 2], 3, 4, [5, [6, 7]]]
flat_data = [1, 2, 3, 4, 5, 6, 7]
What's the best way to do this?
Here's what I have so far but ideally I want everything stuffed into one function:
flat_data = []
def flatten_data(data):
for i in data:
if type(i) == list:
flatten_data(i)
else:
flat_data.append(i)
return flat_a
flatten_data(data)
# [1, 2, 3, 4, 5, 6, 7]
