a3_data_visualisation.py

matplotlib.rcParams.update({'font.size': 14})

f, axarr = plt.subplots(1, 3, figsize=(20, 4))
axarr[0].scatter(df_data["distance_to_city_center"], df_data["price"],
                 edgecolor='black', linewidth='1', s=70, alpha=0.7, c="#e84629")
axarr[0].set_xlabel("Distance to city center [km]")
axarr[0].set_ylabel("Price [k zł]")
axarr[0].set_ylim(0, 1000)
axarr[0].set_yticks(np.arange(0, 1000, 100))
axarr[0].set_xticks(np.arange(0, 13, 1))
axarr[0].grid(color='red', linestyle='--', linewidth=1, alpha=0.2)
axarr[0].spines["top"].set_visible(False)
axarr[0].spines["right"].set_visible(False)
axarr[0].spines["bottom"].set_visible(False)
axarr[0].spines["left"].set_visible(False)

axarr[1].scatter(df_data["rooms"], df_data["price"],
                 edgecolor='black', linewidth='1', s=70, alpha=0.7, c="#7dd815")
axarr[1].set_xlabel("Number of rooms")
axarr[1].set_ylabel("Price [k zł]")
axarr[1].set_ylim(0, 1000)
axarr[1].set_yticks(np.arange(0, 1000, 100))
axarr[1].set_xticks(np.arange(0, 6, 1))
axarr[1].grid(color='green', linestyle='--', linewidth=1, alpha=0.2)
axarr[1].spines["top"].set_visible(False)
axarr[1].spines["right"].set_visible(False)
axarr[1].spines["bottom"].set_visible(False)
axarr[1].spines["left"].set_visible(False)

axarr[2].scatter(df_data["size"], df_data["price"],
                 edgecolor='black', linewidth='1', s=70, alpha=0.7, c="#3176f7")
axarr[2].set_xlabel("Size [m^2]")
axarr[2].set_ylabel("Price [k zł]")
axarr[2].set_ylim(0, 1000)
axarr[2].set_yticks(np.arange(0, 1000, 100));
axarr[2].set_xticks(np.arange(0, 120, 10));
axarr[2].grid(color='blue', linestyle='--', linewidth=1, alpha=0.2)
axarr[2].spines["top"].set_visible(False)
axarr[2].spines["right"].set_visible(False)
axarr[2].spines["bottom"].set_visible(False)
axarr[2].spines["left"].set_visible(False);