test-econ-model/scripts/visualize-pop-results.py

21 lines
626 B
Python

from matplotlib import pyplot as plt
import pandas as pd
with open("../pop-balances-out-inheritance.csv", "r") as file:
pop_inheritance_results = pd.read_csv(file)
plt.hist(pop_inheritance_results.loc[:,"Balance"].sort_values(), bins=30)
plt.title("Wealth distribution (Inheritance)")
plt.xlabel("dollarydoos")
plt.ylabel("participants")
plt.show()
with open("../pop-balances-out-redist.csv", "r") as file:
pop_redist_results = pd.read_csv(file)
plt.hist(pop_redist_results.loc[:,"Balance"].sort_values(), bins=30)
plt.title("Wealth distribution (")
plt.xlabel("dollarydoos")
plt.ylabel("participants")
plt.show()