用Python绘制点阵树

时间:2015-11-14 19:02:05

标签: python python-3.x binary-tree hierarchical-data mathematical-lattices

我正在寻找将元组树t = ((4,), (3, 5,), (2, 4, 6,), (1, 3, 5, 7,))绘制为下图的想法(假设这个二叉树的大小可以改变)。我试图避免依赖非核心软件包(只是坚持使用pandas,numpy,matplotlib,scikit等)。

enter image description here

1 个答案:

答案 0 :(得分:1)

我正在使用这段代码,它给出了很好的结果:

from matplotlib import pyplot as plt
import numpy as np

fig = plt.figure(figsize=[5, 5])
for i in range(3):
    x = [1, 0, 1]
    for j in range(i):
        x.append(0)
        x.append(1)
    x = np.array(x) + i
    y = np.arange(-(i+1), i+2)[::-1]
    plt.plot(x, y, 'bo-')
plt.show()

enter image description here