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

49次阅读
没有评论

共计 547 个字符,预计需要花费 2 分钟才能阅读完成。

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

丸趣 TV 网 – 提供最优质的资源集合!

正文完
 
丸趣
版权声明:本站原创文章,由 丸趣 2024-04-16发表,共计547字。
转载说明:除特殊说明外本站除技术相关以外文章皆由网络搜集发布,转载请注明出处。
评论(没有评论)