Replication of the Russian perevertysh

_murom.py

#!/usr/bin/env python3

from matplotlib  import pyplot as plt
from numpy       import arange

def find(func, iterable):
	result = None
	for i, elem in enumerate(iterable):
		if func(elem):
			assert result is None
			result = i
	return result

def load(filename):
	with open(filename, 'r') as file:
		data = [line.split('\t') for line in file]
	tik = find(lambda s: s == 'tik', data[0])
	uik = find(lambda s: s == 'uik', data[0])
	inv = find(lambda s: ' недействительных ' in s, data[0])
	val = find(lambda s: ' действительных ' in s, data[0])
	orl = find(lambda s: s.startswith('Орлова'), data[0])
	sip = find(lambda s: s.startswith('Сипягин'), data[0])

	return [(row[uik],
	         int(row[orl]) / (int(row[val]) + int(row[inv])),
	         int(row[sip]) / (int(row[val]) + int(row[inv])))
	        for row in data
	        if len(row) == len(data[0])
	        if row[tik].endswith('Муром округ')]

fst = load('first.tsv')
snd = load('second.tsv')
assert all(f[0] == s[0] for f, s in zip(fst, snd))

results = [f + s[1:] for f, s in zip(fst, snd)]
# 0 = УИК, 1 = Орлова 1-й, 2 = Сипягин 1-й, 3 = Орлова 2-й, 4 = Сипягин 2-й

index = arange(len(results))
width = 1/3

def project(iterable, index):
	return [elem[index] for elem in iterable]

def plot(ax, orlovas, sipyagins):
	ax.bar(index, orlovas, width, color='b', label='Орлова')
	ax.bar(index + width, sipyagins, width, color='y', label='Сипягин')
	ax.set_ylim(0.0, 1.0)
	ax.set_xticks([])

fig, [topax, botax] = plt.subplots(nrows=2)
plot(topax, project(results, 1), project(results, 2))
plot(botax, project(results, 3), project(results, 4))

topax.legend()
topax.set_ylabel('I тур')
botax.set_ylabel('II тур')
botax.set_xlabel('Участок')
botax.set_xticks(index + width/2)
botax.set_xticklabels(project(results, 0), rotation='vertical')

fig.suptitle('Выборы губернатора Владимирской области, Муром')
fig.tight_layout()
plt.show()

first.tsv

second.tsv