面试题2：Flatten Arrays 数组扁平化

最新推荐文章于 2025-11-24 01:18:54 发布

原创最新推荐文章于 2025-11-24 01:18:54 发布 · 365 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #算法 #面试

本文介绍了一种将多维数组转换为一维数组的方法——数组扁平化，并提供了两种实现方式：一种返回列表，另一种返回迭代器。这两种方法都通过递归遍历多维数组的每个元素来实现。

数组扁平化，就是将一个多维数组转为一个一维数组。实现方法很简单，递归即可：

from collections.abc import Iterable


# return list
def flatten(input_arr, output_arr=None):
    if output_arr is None:
        output_arr = []
    for ele in input_arr:
        if not isinstance(ele, str) and isinstance(ele, Iterable):
            flatten(ele, output_arr)    #tail-recursion
        else:
            output_arr.append(ele)      #produce the result
    return output_arr


# returns iterator
def flatten_iter(iterable):
    """
    Takes as input multi dimensional iterable and
    returns generator which produces one dimensional output.
    """
    for element in iterable:
        if not isinstance(element, str) and isinstance(element, Iterable):
            yield from flatten_iter(element)    
        else:
            yield element