平面列表
1.问题描述
给定一个列表,该列表中有的元素是列表,有的元素是整数,将其变成只包含整数的简单列表
2.问题示例
输入[[1, 1], 2, [3, 4]],输出[1, 1, 2, 3, 4]
def plane_list(a_list): ret_list = [] i, i_length = 0, len(a_list) while i < i_length: if isinstance(a_list[i], list): j, j_length = 0, len(a_list[i]) while j < j_length: ret_list.append(a_list[i][j]) j += 1 elif isinstance(a_list[i], int): ret_list.append(a_list[i]) else: print('not define') i += 1 return ret_list a_list = [[1, 1], 2, [3, 4]] print(plane_list(a_list))