共计 547 个字符,预计需要花费 2 分钟才能阅读完成。
在 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 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 网 – 提供最优质的资源集合!
正文完