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()