树

tree 是一个用于处理嵌套数据结构的库。从某种意义上说，tree 对 Python 内置的 map 函数进行了泛化——后者仅支持扁平化的序列——而 tree 则允许将一个函数应用于每个“叶节点”，同时保持整体的数据结构不变。

>>> import tree
>>> structure = [[1], [[[2, 3]]], [4]]
>>> tree.flatten(structure)
[1, 2, 3, 4]
>>> tree.map_structure(lambda v: v**2, structure)
[[1], [[[4, 9]]], [16]]

tree 的底层实现基于高效的 C++ 代码，非常适合在高要求的应用场景中使用，例如机器学习模型。

安装

通过 PyPI 安装：

$ pip install dm-tree

直接从 GitHub 使用 pip 安装：

$ pip install git+git://github.com/deepmind/tree.git

从源码构建：

$ python setup.py install

支持

如果您遇到任何问题，请通过在我们的 issue tracker 上提交 issue 来告知我们。

许可证

该项目采用 Apache 2.0 许可证授权。

Tree 快速上手指南

tree 是一个用于处理嵌套数据结构的 Python 库。它扩展了内置的 map 函数（仅支持扁平序列），允许用户对嵌套结构中的每个“叶子节点”应用函数，同时保持整体结构不变。该库底层由优化的 C++ 实现支撑，非常适合机器学习等高性能场景。

环境准备

• 操作系统：Linux、macOS 或 Windows
• Python 版本：Python 3.6 及以上
• 前置依赖：
  • pip 包管理工具
  • 编译环境（若从源码构建，需安装 C++ 编译器，如 gcc 或 Visual Studio Build Tools）

安装步骤

推荐通过 PyPI 安装，国内用户可使用清华镜像源加速下载：

$ pip install dm-tree -i https://pypi.tuna.tsinghua.edu.cn/simple

或者直接从 GitHub 安装：

$ pip install git+git://github.com/deepmind/tree.git

如需从源码构建：

$ git clone https://github.com/deepmind/tree.git
$ cd tree
$ python setup.py install

基本使用

以下示例展示如何展平嵌套结构以及对叶子节点进行映射操作：

>>> import tree
>>> structure = [[1], [[[2, 3]]], [4]]
>>> tree.flatten(structure)
[1, 2, 3, 4]
>>> tree.map_structure(lambda v: v**2, structure)
[[1], [[[4, 9]]], [16]]

• tree.flatten()：将嵌套结构展平为列表。
• tree.map_structure(func, structure)：对结构中每个叶子节点应用 func，返回同结构的新数据。