我们说我有以下列表data
,我想列出flat_data
列表:
data = [[1, 2], 3, 4, [5, [6, 7]]]
flat_data = [1, 2, 3, 4, 5, 6, 7]
最好的方法是什么?
这是我到目前为止所做的事情,但理想情况下我想把所有东西都塞进一个功能中:
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]