Commit 36beaeba authored by leoborn's avatar leoborn
Browse files

Added plotting script

parent ab9fbd4c
Loading
Loading
Loading
Loading

src/plot_gender.py

0 → 100644
+30 −0
Original line number Diff line number Diff line
import pandas as pd
import seaborn as sns; sns.set()
import matplotlib.pyplot as plt
import argparse

parser = argparse.ArgumentParser(description='Plot gender/time/punishment distribtuin')
parser.add_argument('punishment', type=str, 
        choices=["VERDICT=guilty",
            "death",
            "VERDICT=notGuilty",
            "transport",
            "imprison",
            "corporal",
            "miscPunish",
            "noPunish"],
        help='punishment')

args = parser.parse_args()
gender = pd.read_csv('gender_punishment_time.csv',sep="\t")

print(gender)

g=sns.lineplot(data=gender,hue="gender",x="decade",y=args.punishment)
#g=sns.lineplot(data=gender,hue="gender",x="decade",y="VERDICT=guilty")

#g=sns.barplot(data=gender,x="decade",y="VERDICT=notGuilty")
#g=sns.barplot(data=gender,x="decade",y="VERDICT=guilty")


g.get_figure().savefig(args.punishment+"-gender-time.png")