finished tast 1

This commit is contained in:
Kristofers Solo 2023-12-23 15:23:26 +02:00
parent 651f0764d8
commit 06693d328f

48
main.py
View File

@ -3,6 +3,8 @@
from pathlib import Path from pathlib import Path
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import numpy as np
import pandas as pd import pandas as pd
from loguru import logger from loguru import logger
@ -26,35 +28,37 @@ def read_data(path: Path) -> pd.DataFrame:
return dataframe return dataframe
# def visualize() -> None: def bar_chart() -> None:
# df_wind_speed = read_data(WIND_SPEED_PATH) df_wind_speed = read_data(WIND_SPEED_PATH).mean(axis=1)
# df_wind_gusts = read_data(WIND_GUSTS_PATH) df_wind_gusts = read_data(WIND_GUSTS_PATH).max(axis=1) - df_wind_speed
# index = range(len(df_wind_speed))
# bar_width = 0.35
#
# for column, (mean_speed, max_speed) in enumerate(zip(df_wind_speed.columns, df_wind_gusts.columns)):
# plt.bar(index, df_wind_speed[column], width=bar_width, label=f"Vidējais vēja ātrums {mean_speed}", color="blue")
# plt.bar(index, df_wind_gusts[column], width=bar_width, label=f"Vēja ātrums brāzmās {mean_speed}", color="orange")
#
# plt.figure(figsize=(12, 6))
# plt.xlabel("Mērījumu Datums")
# plt.ylabel("Vēja ātrums (m/s)")
# plt.title("Vidējais un maksimālais vēja ātrums 2023. gada augustā")
# plt.legend()
# plt.show()
df_combined = pd.concat(
[df_wind_speed, df_wind_gusts],
axis=1,
)
def task2() -> None: df_combined.columns = ["Vidējais", "Maksimālais"]
# create a bar chart df_combined.plot.bar(stacked=True, figsize=(12, 8), color=["#ff7f0e", "#1f77b4"], width=0.6)
df_air_temp = read_data(AIR_TEMP_PATH)
plt.bar(df_air_temp, height=10, width=0.8) plt.yticks(np.arange(0, df_combined.max().max() + 2.5, 2.5))
plt.xticks(rotation=45)
plt.xlabel("Mērījumu Datums")
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:
# # create a bar chart
# df_air_temp = read_data(AIR_TEMP_PATH)
# plt.bar(df_air_temp, height=10, width=0.8)
# plt.show()
@logger.catch @logger.catch
def main() -> None: def main() -> None:
# visualize() bar_chart()
task2()
if __name__ == "__main__": if __name__ == "__main__":