Ejemplo n.º 1
0
def test_two_bar_with_legend():
    fig = plt.figure()
    ax = fig.add_subplot(111)

    np.random.seed(0)

    x = np.arange(6)
    y1 = np.random.uniform(size=6) * 100
    y2 = np.random.uniform(size=6) * 100

    mpt.bar(ax, x, y1, width=0.4, label="Data 1", annotations=[str(int(d)) for d in y1])
    mpt.bar(ax, x + 0.4, y2, width=0.4, label="Data 2", annotations=[str(int(d)) for d in y2])
    ax.set_xticks(np.arange(6) + 0.4)
    ax.set_xticklabels(["Set {index}".format(index=str(i)) for i in np.arange(6) + 1])
    mpt.legend(ax)
Ejemplo n.º 2
0
def test_stack_bar_with_legend():
    fig = plt.figure()
    ax = fig.add_subplot(111)

    np.random.seed(0)

    x = np.arange(6)
    y1 = -np.random.uniform(size=6) * 50
    y2 = np.random.uniform(size=6) * 50
    y3 = np.random.uniform(size=6) * 50

    mpt.bar(
        ax,
        x,
        y1,
        label="Data 1",
        annotations=[str(int(d)) for d in y1],
        annotations_loc="center",
        orientation="horizontal",
        grid="True",
        ticks=["Set {index}".format(index=str(i)) for i in np.arange(6) + 1],
    )
    mpt.bar(
        ax, x, y2, label="Data 2", annotations=[str(int(d)) for d in y2], annotations_loc="in", orientation="horizontal"
    )
    mpt.bar(
        ax,
        x,
        y3,
        offset=y2,
        label="Data 3",
        annotations=[str(int(d)) for d in y2 + y3],
        annotations_loc="out",
        orientation="horizontal",
    )
    mpt.legend(ax, loc="center right")
Ejemplo n.º 3
0
def content(fig):
    ax = fig.add_subplot(321)

    np.random.seed(0)

    for i in range(8):
        y = np.random.normal(size=1000).cumsum()
        x = np.arange(1000)

        mpt.plot(ax, x, y, label='Data {index}'.format(index=str(i + 1)))
    mpt.legend(ax, ncol=2, loc='lower left')

    ax = fig.add_subplot(322)

    np.random.seed(0)

    x = np.arange(6)
    y1 = np.random.uniform(size=6) * 100
    y2 = np.random.uniform(size=6) * 100

    mpt.bar(ax, x, y1, width=0.4, label='Data 1',
            annotations=[str(int(d)) for d in y1])
    mpt.bar(ax, x + 0.4, y2, width=0.4, label='Data 2',
            annotations=[str(int(d)) for d in y2])
    ax.set_xticks(np.arange(6) + 0.4)
    ax.set_xticklabels(
        ['Set {index}'.format(index=str(i)) for i in np.arange(6) + 1])
    mpt.legend(ax)

    ax = fig.add_subplot(323)

    np.random.seed(1)

    for i in range(8):
        x = np.random.normal(size=1000) + i

        mpt.hist(ax, x, label='Data {index}'.format(index=str(i + 1)),
                 orientation='horizontal', alpha=0.6)

    ax = fig.add_subplot(324)

    np.random.seed(0)

    x = np.random.normal(size=(1000, 8))

    mpt.boxplot(
        ax, x, ticks=['Set {index}'.format(index=i + 1) for i in range(8)])

    ax = fig.add_subplot(325)

    np.random.seed(0)

    for i in range(8):
        x = np.random.normal(size=1000) + i
        y = np.random.normal(size=1000) + i

        mpt.scatter(ax, x, y, label='Data {index}'.format(index=str(i + 1)))
    mpt.legend(ax, title='Legend', scatterpoints=3)

    ax = fig.add_subplot(326)

    np.random.seed(0)

    mpt.pcolormesh(ax, np.random.rand(20, 20), xticks=[str(i) for i in range(20)],
                   yticks=[str(i + 20) for i in range(20)])
Ejemplo n.º 4
0
import numpy as np
import matplotlib.pylab as plt
import matplottheme as mpt

np.random.seed(0)

x = np.random.rand(5) * 100

fig, ax = plt.subplots()

mpt.bar(ax, np.arange(5), x)

plt.savefig('bar.png', transparent=True)