数组扁平化+ 实现一个flat带层数

**数组扁平化与实现Flat函数**

在编程中，数组扁平化是指将多维数组转换为一维数组的过程。这种操作对于数据处理、分析和展示至关重要。在本文中，我们将讨论如何实现一个`flat`函数来进行数组扁平化，并且支持层数参数。

### 数组扁平化首先，让我们了解一下什么是数组扁平化。假设我们有一个二维数组，如下所示：

array = [
 [1,2,3],
 [4,5,6],
 [7,8,9]
]

flat_array = [1,2,3,4,5,6,7,8,9]

def flat(array, depth=1):
 """
 将多维数组转换为一维数组。
 Args:
 array (list): 需要扁平化的数组。
 depth (int, optional): 扁平化的层数。 Defaults to1.
 Returns:
 list: 扁平化后的数组。
 """
 # 如果depth为0，直接返回原数组 if depth ==0:
 return array # 初始化结果列表 result = []
 # 遍历每个元素 for element in array:
 # 如果元素是列表，则递归扁平化 if isinstance(element, list):
 result.extend(flat(element, depth -1))
 else:
 # 否则，将其添加到结果列表中 result.append(element)
 return result

# 示例1：扁平化一个二维数组array = [
 [1,2,3],
 [4,5,6],
 [7,8,9]
]

print(flat(array)) # 输出：[1,2,3,4,5,6,7,8,9]

# 示例2：扁平化一个三维数组array = [
 [[1,2], [3,4]],
 [[5,6], [7,8]]
]

print(flat(array)) # 输出：[1,2,3,4,5,6,7,8]

# 示例3：指定层数array = [
 [1,2],
 [3,4]
]

print(flat(array, depth=2)) # 输出：[[1,2], [3,4]]