How can I visualize binary trees in Python?

How can I visualize binary trees in Python?

You can use libraries like Matplotlib or Graphviz to visualize binary trees in Python. Would you like a code example?

Yes, please provide a code example using Matplotlib.

Sure! Here's a simple example:

import matplotlib.pyplot as plt
import networkx as nx

class Node:
    def __init__(self, value):
        self.value = value
        self.left = None
        self.right = None

def add_node(root, value):
    if root is None:
        return Node(value)
    if value < root.value:
        root.left = add_node(root.left, value)
    else:
        root.right = add_node(root.right, value)
    return root

# Visualization function
def visualize_tree(node, graph=None, pos=None, parent=None, parsed=[]):
    if graph is None:
        graph = nx.DiGraph()
    if node is not None:
        graph.add_node(node.value)
        if parent is not None:
            graph.add_edge(parent, node.value)
        pos[node.value] = (len(parsed), -len(parsed))
        parsed.append(node.value)
        visualize_tree(node.left, graph, pos, node.value, parsed)
        visualize_tree(node.right, graph, pos, node.value, parsed)
    return graph, pos

# Example usage
root = add_node(None, 10)
add_node(root, 5)
add_node(root, 15)

graph, pos = visualize_tree(root)

nx.draw(graph, pos, with_labels=True)
plt.show()

This is great! Can I customize the appearance of the tree?

Yes, you can customize the appearance by changing node colors, sizes, and edge styles in Matplotlib. Would you like to see how?