plot_intervention_distr.py
14482acae4e2master

plot_intervention_distr.py
View Options

			import pandas as pd
			import matplotlib.pyplot as plt
			import math


			def plot(path):
			""" Plots the distribution of interventions per party per meeting.

			Args:
			path (str): path to the interventions_prepared.csv file
			"""

			data = pd.read_csv(path, encoding="utf-8-sig")
			plt.hist(data["nb_interventions"], bins=40)
			# plt.hist((data["nb_interventions"].loc[data["nb_interventions"] > 0]).map(lambda n: math.log(n+10)).map(lambda n: math.log(n+10)), bins=40)
			plt.title("Distribution of interventions")
			plt.ylabel("Occurences")
			plt.xlabel("Nb. of interventions")
			plt.show()
			# data["meeting_and_country"] = str(data["meeting"]) + str(data["country"])

			"""by_nb_interventions = data.groupby("nb_interventions")

			result_df = pd.DataFrame(columns={"nb_interventions", "nb_affiliations"})

			for n, aff in by_nb_interventions:
			result_df = result_df.append({
			"nb_interventions": n,
			"nb_affiliations": len(aff)
			}, ignore_index=True)

			plot_data = result_df.set_index("nb_interventions")
			plot_data.plot(kind="hist", xlabel="nb of interventions", ylabel="number of affiliations",
			title="Distribution of number of interventions per party and meeting", xlim=[-1000,1000])
			plt.show()
			"""