import matplotlib.pyplot as plt import numpy as np from market import Market # Baseline ad, bd, az, bz, tax baseline_params = 15, .5, -2, .5, 3 m = Market(*baseline_params) q_max = m.quantity() * 2 q_grid = np.linspace(0.0, q_max, 100) pd = m.inverse_demand(q_grid) ps = m.inverse_supply(q_grid) psno = m.inverse_supply_no_tax(q_grid) fig, ax = plt.subplots() ax.plot(q_grid, pd, lw=2, alpha=0.6, label='demand') ax.plot(q_grid, ps, lw=2, alpha=0.6, label='supply') ax.plot(q_grid, psno, '--k', lw=2, alpha=0.6, label='supply without tax') ax.set_xlabel('quantity', fontsize=14) ax.set_xlim(0, q_max) ax.set_ylabel('price', fontsize=14) ax.legend(loc='lower right', frameon=False, fontsize=14) plt.show()