maphel-langtime/plot_grid_diversities_figure5.py at main · DigitalGeographyLab/maphel-langtime · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
# -*- coding: utf-8 -*-
"""
Created on Tue Mar 28 11:55:09 2023

@author: TuoVaisanen-e01
"""
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np
import argparse as ap

# set up argument parser
ap = argparse.ArgumentParser()

# Get grid file
ap.add_argument("-sp", "--shannonpop", required=True,
                help="Path to file containing population-weighted Shannon entropies in neighbourhoods")

# Get path to input file
ap.add_argument("-up", "--uniquepop", required=True,
                help="Path to file containing population-weighted unique languages in neighbourhoods")

# Get path to input file
ap.add_argument("-sn", "--shannorm", required=True,
                help="Path to file with normalized Shannon entropies in neighbourhoods")

# Get path to input file
ap.add_argument("-un", "--uniquenorm", required=True,
                help="Path to file with normalized unique language counts in neighbourhoods")

# Get path to output file
ap.add_argument("-o", "--output", required=True,
                help="Path to output folder. For example: /path/to/folder/. This script assumes you have access to FOLK data within Fiona")

# parse arguments
args = vars(ap.parse_args())

# read data in
shannorm = pd.read_pickle(args['shannorm'])
uniqnorm = pd.read_pickle(args['uniquenorm'])
shanpopw = pd.read_pickle(args['shannonpop'])
uniqpopw = pd.read_pickle(args['uniquepop'])

# drop neighbourhoods where population groups have left
shanpopw = shanpopw[shanpopw['celltype'] != 'Ceased']
uniqpopw = uniqpopw[uniqpopw['celltype'] != 'Ceased']
shannorm = shannorm[shannorm['type'] != 'Ceased grid cells']
uniqnorm = uniqnorm[uniqnorm['type'] != 'Ceased grid cells']

# set theme
sns.set()
sns.set(font_scale=1.2)

# start plotting
fig, ax = plt.subplots(2,2, figsize=(15,14))
ax = ax.flatten()

# set palette
palette = {'Somali-inhabited':'C1','Estonian-inhabited':'C0','Finnish-inhabited':'C3'}

# plot unique languages
g = sns.lineplot(x='year', y='score', hue='type', style='celltype', data=uniqpopw,
                 ci=99, estimator=np.nanmean, n_boot=3000, palette=palette, ax=ax[0])
g.set(xlabel='', ylabel='Unique languages', ylim=(0,40))
g.set_title('a.', loc='left', fontsize=15)
handles = ax[0].get_legend().legendHandles
ax[0].get_legend().remove()
ax[0].legend(handles, ['Residential neighbourhood', 'Somali-inhabited','Estonian-inhabited', 'Finnish-inhabited',
                    '\nTemporal type','Present every year', 'New grid cells'])

# plot normalized unique languages
g = sns.lineplot(x='year', y='unique_langs', hue='gridtype', style='type', data=uniqnorm,
                 ci=99, estimator=np.nanmean, n_boot=3000, palette=palette, ax=ax[1],
                 legend=False)
g.set(xlabel='', ylabel='Normalized unique languages')
g.set_title('b.', loc='left', fontsize=15)

# plot shannon entropy
g = sns.lineplot(x='year', y='score', hue='type', style='celltype', data=shanpopw,
                 ci=99, estimator=np.nanmean, n_boot=3000, palette=palette, ax=ax[2],
                 legend=False)
g.set(xlabel='', ylabel='Shannon entropy', ylim=(0,1.9))
g.set_title('c.', loc='left', fontsize=15)

# plot normalized shannon entropy
g = sns.lineplot(x='year', y='shannon', hue='gridtype', style='type', data=shannorm,
                 ci=99, estimator=np.nanmean, n_boot=3000, palette=palette, ax=ax[3],
                 legend=False)
g.set(xlabel='', ylabel='Normalized Shannon entropy')
g.set_title('d.', loc='left', fontsize=15)

# save the figure
plt.savefig(args['output'] + 'figure5_grid_new_old_trajectories.pdf', dpi=300, bbox_inches='tight')