二维列表转一维列表

# 使用 sum 函数
arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
res = sum(arr, [])
print(res)

# 使用标准库 itertools chain 方法
import itertools
res = list(itertools.chain(*arr))
print(res)

# 使用递归
arrs = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9, [10, 11, 12]]
]

def change_arr(arrs: list, temp: list) -> list:
    for arr in arrs:
        if not isinstance(arr, list):
            temp.append(arr)
        else:
            change_arr(arr, temp)
    return temp

res = []
res = change_arr(arrs, res)
print(res)

你可能感兴趣的:(python,python爬虫,python,二维列表,sum,chain,itertools)