kristoferssolo/LU-Data-Visualisation
1
0
Fork 0
mirror of https://github.com/kristoferssolo/LU-Data-Visualisation.git synced 2025-10-21 20:10:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: determine season

Browse Source
This commit is contained in:
Kristofers Solo 2023-12-24 16:05:44 +02:00
parent 06693d328f
commit c1d4a4b796
1 changed files with 38 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
main.py
View File

@ -23,17 +23,29 @@ AIR_TEMP_PATH = BASE_PATH.joinpath("data", "gaisaTemperatura2022.xlsx")
def read_data(path: Path) -> pd.DataFrame: def read_data(path: Path) -> pd.DataFrame:
dataframe = pd.read_excel(path, parse_dates=["Datums"]) dataframe = pd.read_excel(path, date_parser="Datums", index_col="Datums")
dataframe.set_index("Datums", inplace=True)
return dataframe return dataframe
def get_season(month: int) -> str | None:
if month in [12, 1, 2]:
return "Ziema"
elif month in [3, 4, 5]:
return "Pavasaris"
elif month in [6, 7, 8]:
return "Vasara"
elif month in [9, 10, 11]:
return "Rudens"
else:
return None
def bar_chart() -> None: def bar_chart() -> None:
df_wind_speed = read_data(WIND_SPEED_PATH).mean(axis=1) df_avg = read_data(WIND_SPEED_PATH).mean(axis=1)
df_wind_gusts = read_data(WIND_GUSTS_PATH).max(axis=1) - df_wind_speed df_max = read_data(WIND_GUSTS_PATH).max(axis=1) - df_avg
df_combined = pd.concat( df_combined = pd.concat(
[df_wind_speed, df_wind_gusts], [df_avg, df_max],
axis=1, axis=1,
) )
@ -43,22 +55,36 @@ def bar_chart() -> None:
plt.yticks(np.arange(0, df_combined.max().max() + 2.5, 2.5)) plt.yticks(np.arange(0, df_combined.max().max() + 2.5, 2.5))
plt.xticks(rotation=45) plt.xticks(rotation=45)
plt.title("Vidējais un maksimālais vēja ātrums 2023. gada augustā")
plt.xlabel("Mērījumu Datums") plt.xlabel("Mērījumu Datums")
plt.ylabel("Vēja ātrums (m/s)") plt.ylabel("Vēja ātrums (m/s)")
plt.title("Vidējais un maksimālais vēja ātrums 2023. gada augustā")
plt.show() plt.show()
# def task2() -> None: SEASONS = {
# # create a bar chart 1: "Ziema",
# df_air_temp = read_data(AIR_TEMP_PATH) 2: "Pavasaris",
# plt.bar(df_air_temp, height=10, width=0.8) 3: "Vasara",
# plt.show() 4: "Rudens",
}
def box_plot() -> None:
df = read_data(AIR_TEMP_PATH)
df.index = pd.to_datetime(df.index, format="%d.%m.%Y")
df["Season"] = df.index.month % 12 // 3 + 1
df["Season"] = df["Season"].map(SEASONS)
plt.title("Gaisa temperatūra Rīgā četros gadalaikos")
plt.ylabel("Gaisa temperatūra (Celsija grādos)")
# plt.show()
@logger.catch @logger.catch
def main() -> None: def main() -> None:
bar_chart() # bar_chart()
box_plot()
if __name__ == "__main__": if __name__ == "__main__":