Skip to content

Commit a904aaa

Browse files
committed
Fix pandas warnings
1 parent 3615841 commit a904aaa

File tree

2 files changed

+3
-7
lines changed

2 files changed

+3
-7
lines changed

src/stravavis/plot_dumbbell.py

+1-5
Original file line numberDiff line numberDiff line change
@@ -40,11 +40,7 @@ def plot_dumbbell(
4040

4141
# Get activity start and end times
4242
activities["start"] = activities["Activity Date"]
43-
activities["duration"] = 0
44-
for i in range(len(activities)):
45-
activities["duration"][i] = pd.Timedelta(
46-
activities["Elapsed Time"][i], unit="s"
47-
)
43+
activities["duration"] = pd.to_timedelta(activities["Elapsed Time"], unit="s")
4844
activities["end"] = pd.to_datetime(activities["start"] + activities["duration"])
4945

5046
# Remove activities outside the year_min -> year_max window

src/stravavis/plot_landscape.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@ def plot_landscape(df, output_file="landscape.png"):
1919
processed = []
2020

2121
for activity in track(activities, "Processing tracks"):
22-
df_i = df[df["name"] == activity]
23-
df_i["dist_norm"] = (df_i["dist"] - df_i["dist"].min()) / (
22+
df_i = df[df["name"] == activity].copy()
23+
df_i.loc[:, "dist_norm"] = (df_i["dist"] - df_i["dist"].min()) / (
2424
df_i["dist"].max() - df_i["dist"].min()
2525
)
2626
processed.append(df_i)

0 commit comments

Comments
 (0)