python 如何将一个多维列表中的元素一个个按顺序按顺序取出

首先声明,这种问题的正统解法是使用numpy

下面是不考虑效率的 通用版本,满足 "python自带" 这个条件, 并且可自定义 过滤条件 与 get函数

import json
import ast

def flat(nest,cond_func=lambda r:type(r)==ast.Num,get_func=lambda r:r.n):
    cd = json.dumps(nest)
    t = ast.parse(cd)
    g = ast.walk(t)
    arr = list(g)
    arr = list(filter(cond_func,arr))
    arr = list(map(get_func,arr))
    return(arr)

arr = [1, [4], 2, [3, [5, 6]]]
flat(arr)
# [1, 2, 4, 3, 5, 6]

cubic = [
    [[255, 0, 0], [0, 255, 0], [0, 0, 255]], 
    [[255, 0, 0], [0, 255, 0], [0, 0, 255]], 
    [[255, 0, 0], [0, 255, 0], [0, 0, 255]]
]

flat(cubic)
#[255, 0, 0, 0, 255, 0, 0, 0, 255, 255, 0, 0, 0, 255, 0, 0, 0, 255, 255, 0, 0, 0, 255, 0, 0, 0, 255]

arr = [1, ["a"], 2, ["b", [5, "c"]]]
flat(arr,cond_func=lambda r:type(r)==ast.Str,get_func=lambda r:r.s)
#['a', 'b', 'c']
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值