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

本文介绍了一种不依赖numpy的Python自定义方法,用于扁平化复杂数据结构,通过`json`和`ast`模块解析,可灵活设置过滤条件和获取函数。实例演示了如何处理列表嵌套和字符串过滤。
部署运行你感兴趣的模型镜像

首先声明,这种问题的正统解法是使用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']

您可能感兴趣的与本文相关的镜像

Python3.8

Python3.8

Conda
Python

Python 是一种高级、解释型、通用的编程语言,以其简洁易读的语法而闻名,适用于广泛的应用,包括Web开发、数据分析、人工智能和自动化脚本

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值