Skip to content
/ bigtree Public

Tree Implementation and Methods for Python, integrated with list, dictionary, pandas and polars DataFrame.

License

Notifications You must be signed in to change notification settings

kayjan/bigtree

Repository files navigation

Big Tree Python Package

Tree Implementation and Methods for Python, integrated with list, dictionary, pandas and polars DataFrame.

It is pythonic, making it easy to learn and extendable to many types of workflows.


Related Links:


Components

There are 3 segments to Big Tree consisting of Tree, Binary Tree, and Directed Acyclic Graph (DAG) implementation.

For Tree implementation, there are 9 main components.

  1. 🌺 Node
    1. BaseNode, extendable class
    2. Node, BaseNode with node name attribute
  2. ✨ Constructing Tree
    1. From Node, using parent and children constructors
    2. From str, using tree display or Newick string notation
    3. From list, using paths or parent-child tuples
    4. From nested dictionary, using path-attribute key-value pairs or recursive structure
    5. From pandas DataFrame, using paths or parent-child columns
    6. From polars DataFrame, using paths or parent-child columns
    7. Add nodes to existing tree using path string
    8. Add nodes and attributes to existing tree using dictionary, pandas DataFrame, or polars DataFrame, using path
    9. Add only attributes to existing tree using dictionary, pandas DataFrame, or polars DataFrame, using node name
  3. ➰ Traversing Tree
    1. Pre-Order Traversal
    2. Post-Order Traversal
    3. Level-Order Traversal
    4. Level-Order-Group Traversal
    5. ZigZag Traversal
    6. ZigZag-Group Traversal
  4. 📝 Modifying Tree
    1. Copy nodes from location to destination
    2. Shift nodes from location to destination
    3. Shift and replace nodes from location to destination
    4. Copy nodes from one tree to another
    5. Copy and replace nodes from one tree to another
  5. 🔍 Tree Search
    1. Find multiple nodes based on name, partial path, relative path, attribute value, user-defined condition
    2. Find single nodes based on name, partial path, relative path, full path, attribute value, user-defined condition
    3. Find multiple child nodes based on user-defined condition
    4. Find single child node based on name, user-defined condition
  6. 🔧 Helper Function
    1. Cloning tree to another Node type
    2. Get subtree (smaller tree with different root)
    3. Prune tree (smaller tree with same root)
    4. Get difference between two trees
  7. 📊 Plotting Tree
    1. Enhanced Reingold Tilford Algorithm to retrieve (x, y) coordinates for a tree structure
    2. Plot tree using matplotlib (optional dependency)
  8. 🔨 Exporting Tree
    1. Print to console, in vertical or horizontal orientation
    2. Export to Newick string notation, dictionary, nested dictionary, pandas DataFrame, or polars DataFrame
    3. Export tree to dot (can save to .dot, .png, .svg, .jpeg files)
    4. Export tree to Pillow (can save to .png, .jpg)
    5. Export tree to Mermaid Flowchart (can display on .md)
  9. ✔️ Workflows
    1. Sample workflows for tree demonstration!

For Binary Tree implementation, there are 3 main components. Binary Node inherits from Node, so the components in Tree implementation are also available in Binary Tree.

  1. 🌿 Node
    1. BinaryNode, Node with binary tree rules
  2. ✨ Constructing Binary Tree
    1. From list, using flattened list structure
  3. ➰ Traversing Binary Tree
    1. In-Order Traversal

For Directed Acyclic Graph (DAG) implementation, there are 4 main components.

  1. 🌼 Node
    1. DAGNode, extendable class for constructing Directed Acyclic Graph (DAG)
  2. ✨ Constructing DAG
    1. From list, containing parent-child tuples
    2. From nested dictionary
    3. From pandas DataFrame
  3. ➰ Traversing DAG
    1. Generic traversal method
  4. 🔨 Exporting DAG
    1. Export to list, dictionary, or pandas DataFrame
    2. Export DAG to dot (can save to .dot, .png, .svg, .jpeg files)

Installation

bigtree requires Python 3.8 . There are two ways to install bigtree, with pip (recommended) or conda.

a) Installation with pip

Basic Installation

To install bigtree, run the following line in command prompt:

$ pip install bigtree

Installing optional dependencies

bigtree have a number of optional dependencies, which can be installed using "extras" syntax.

$ pip install 'bigtree[extra_1, extra_2]'

Examples of extra packages include:

  • all: include all optional dependencies
  • image: for exporting tree to image
  • matplotlib: for plotting trees
  • pandas: for pandas methods
  • polars: for polars methods

For image extra dependency, you may need to install more plugins.

$ brew install gprof2dot  # for MacOS
$ conda install graphviz  # for Windows

b) Installation with conda

To install bigtree with conda, run the following line in command prompt:

$ conda install -c conda-forge bigtree

Star History

Star History Chart

About

Tree Implementation and Methods for Python, integrated with list, dictionary, pandas and polars DataFrame.

Topics

Resources

License

Security policy

Stars

Watchers

Forks

Sponsor this project

 

Languages