二维转一维数组的方法很多,使用numpy ,sum 好多方法可以实现。
例如:

arr = [[1,2,3],[4,5,6]]
# 方法一
print([item for i in arr for item in i])
# 方法二
from functools import reduce
print(reduce(lambdax,y:x+y,arr))
# 方法三
print(sum(arr,[]))
  • 多维转一维
def exchange_one_arr(item, res:list):
    """多维转一维"""
    if not isinstance(item,list):
        res.append(item)
        return
    for it in item:
        exchange_one_arr(it, res)
    return res

if __name__ == '__main__':
    arr = [[1, 2, [3, [5], [9]]], [4, [5], 6], [0]]
    print(exchange_one_arr(arr, []))
Logo

GitCode AI社区是一款由 GitCode 团队打造的智能助手,AI大模型社区、提供国内外头部大模型及数据集服务。

更多推荐