阅读量:0
在Python中,可以使用递归函数或者使用内置的itertools.chain
模块来对数组进行扁平化处理。
- 使用递归函数:
def flatten_array(arr): result = [] for i in arr: if isinstance(i, list): result.extend(flatten_array(i)) else: result.append(i) return result # 示例 arr = [1, [2, 3], [4, [5, 6]]] flattened_arr = flatten_array(arr) print(flattened_arr) # 输出:[1, 2, 3, 4, 5, 6]
- 使用
itertools.chain
模块:
import itertools def flatten_array(arr): return list(itertools.chain.from_iterable( (flatten_array(i) if isinstance(i, list) else [i] for i in arr) )) # 示例 arr = [1, [2, 3], [4, [5, 6]]] flattened_arr = flatten_array(arr) print(flattened_arr) # 输出:[1, 2, 3, 4, 5, 6]