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

   2024-10-25 3480
核心提示:在Python中,可以使用递归函数或者使用内置的itertools.chain模块来对数组进行扁平化处理。使用递归函数:def flatten_array(arr

在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 itertoolsdef 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]

 
举报打赏
 
更多>同类网点查询
推荐图文
推荐网点查询
点击排行

网站首页  |  关于我们  |  联系方式网站留言    |  赣ICP备2021007278号