import matplotlib pyplot as plt import seaborn as sns from pandas plot

import matplotlib.pyplot as plt
import seaborn as sns
from pandas.plotting import register_matplotlib_converters
register_matplotlib_converters()

def avg_area(df):
    res = pd.np.mean(df['price']/df['floor_area'])
    return res

q1 = pricesdf.merge(estatesdf.loc[(estatesdf['brick'] == True) & (estatesdf['new_building'] == False)])\
    [['ts', 'price','floor_area']].groupby('ts').apply(avg_area)
q2 = pricesdf.merge(estatesdf.loc[estatesdf['panel'] == True])\
    [['ts', 'price','floor_area']].groupby('ts').apply(avg_area)
q3 = pricesdf.merge(estatesdf.loc[estatesdf['new_building'] == True])\
    [['ts', 'price', 'floor_area']].groupby('ts').apply(avg_area)
q4 = pricesdf.merge(estatesdf)\
    [['ts', 'price', 'floor_area']].groupby('ts').apply(avg_area)

q = pd.concat([q1,q2,q3,q4],axis=1)
q.columns= ['cihla','panel','novostavba','vse']
display(q)
sns.set(style='darkgrid')
sns.lineplot(data=q)