python tree包介绍

deepmind 开发的 tree包是一个用于处理嵌套数据结构的库。在某种程度上,tree推广了只支持平面序列的内置映射函数,并允许将一个函数应用于每个“叶”,以保持整体结构。


### 1. 安装
pip install dm-tree

# 激活conda环境后安装
conda install dm-tree

# 苹果系统安装后,导入包(import tree),提示没有此模块

### 2.示例代码
import tree
structure = [[1], [[[2, 3]]], [4]]
tree.flatten(structure)
result = tree.map_structure(lambda v: v**2, structure)
print(result)

seq_lst = ['aaaaddcc','lleevvv']
reverse_seq_lst = tree.map_structure(lambda x: x[::-1], seq_lst)
print(reverse_seq_lst)

参考:

https://github.com/google-deepmind/tree

你可能感兴趣的:(python,linux)