python数组扁平化处理的方法是什么

avatar
作者
筋斗云
阅读量:0

在Python中,可以使用递归函数或者使用内置的itertools.chain模块来对数组进行扁平化处理。

  1. 使用递归函数:
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] 
  1. 使用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] 

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!