#Author: Adrian Hess

import pandas as pandas
import matplotlib.pyplot as plt
import seaborn as sns

#Nazwa pliku z Excela
file = "full2.xlsx"
all_data = pandas.read_excel(file)

#Wybieramy kolumny do stworzenia heatmapy
#W naszym przypadku, wybieramy kolumny transekt i poletka
#oraz kolorujemy względem liczby roślin Galium w danym przedziale terenu
data = all_data.pivot("poletka", "transekt", "Galium")

data.sort_index(level=0, ascending=False, inplace=True)

sns.set(rc = {'figure.figsize':(12,8)})

#Ustalamy parametry i kolor
heatmap = sns.heatmap(data, annot=True,
            cmap="Greens", linecolor='lightgray')

#Ustawiamy tytuł
plt.title('Występowanie Galium aparine na badanym terenie: ', fontsize = 20)
plt.tight_layout()
plt.show()

fig = heatmap.get_figure()
fig.savefig("heatmapa-na-biologie.png")