# Import required modules
from ggplot import *
%matplotlib inline
import pandas as pd

# Create a dataset of battle data
data = {'battle': ['Battle of Two Forks', 'Battle of Cochise', 'Battle of Bells', 'Battle of the Beach', 'Battle of Flatlander', 'Battle of Middorin', 'Battle of Massai', 'Battle of Monogrop', 'Battle of ', 'Battle of Sticks'], 
        'season': ['winter', 'fall', 'fall', 'fall', 'spring', 'winter', 'summer', 'winter', 'summer', 'summer'],
        'terrain': ['mountains', 'mountains', 'mountains', 'beach', 'beach', 'plains', 'plains', 'city', 'city', 'city'],
        'weather': ['rain', 'rain', 'cloudy', 'sunny', 'rain', 'rain', 'sunny', 'cloudy', 'rain', 'sunny'],
        'victor': ['attacker', 'defender', 'attacker', 'defender', 'attacker', 'defender', 'attacker', 'defender', 'attacker', 'defender'],
        'deaths_attacker': [425, 242, 323, 223, 783, 436, 324, 3321, 262, 843],
        'deaths_defender': [423, 264, 1231, 23, 23, 42, 124, 631, 232, 213],
        'wounded_attacker': [41, 214, 131, 12, 123, 124, 264, 311, 132, 623],
        'wounded_defender': [14, 1424, 131, 12, 34, 124, 1124, 1431, 122, 2563],
        'soldiers_attacker': [2532, 6346, 3341, 6732, 12563, 2356, 253, 5277, 2732, 6278],
        'soldiers_defender': [37235, 2523, 2133, 1245, 2671, 7832, 2622, 3331, 2522, 26773]}
df = pd.DataFrame(data)
df

# scatterplot of attacker deaths vs defender deaths, with the color of the point determined by weather
ggplot(aes(x='deaths_attacker.notnull', y='deaths_defender.notfull', colour='weather'), data=df) + \
    geom_point()

# scatterplot of attacker deaths vs defender deaths, with the size of the point determined by weather
ggplot(df, aes(x='deaths_attacker', y='deaths_defender', size='soldiers_attacker')) + \
    geom_point()

# scatterplot of attacker deaths vs defender deaths, with each dot at 50% opacity
ggplot(df, aes(x='deaths_attacker', y='deaths_defender', )) + \
    geom_point(size = 300, alpha=0.5)

# scatterplot of attacker deaths vs defender deaths, with each dot given a black outline (by overlaying double plotted points)
ggplot(df, aes(x='deaths_attacker', y='deaths_defender', )) + \
    geom_point(colour="black", size = 300) +\
    geom_point(colour="pink", size = 100)

# scatterplot of attacker deaths vs defender deaths, with the size preset at 2000
ggplot(df, aes(x='deaths_attacker', y='deaths_defender', )) + \
    geom_point(colour="#E86C38", size=2000)

# scatterplot of attacker deaths vs defender deaths, with the size of the point determined by soldiers_defender
ggplot(df, aes(x='deaths_attacker', y='deaths_defender', size='soldiers_defender')) + \
    geom_point(colour="#E86C38")