回答編集履歴

2024/02/21 16:33

投稿

スコア20675

test CHANGED Viewed

@@ -9,15 +9,16 @@
     '2020-01-01 12:00:00']
 s1 = pd.to_datetime(pd.Series(date_strs), format='%Y-%m-%d %H:%M:%S', errors='coerce')
-gid = s1.diff(-1).dt.total_seconds().eq(0).cumsum()
+gid = (s1.notna() & s1.diff().dt.total_seconds().ne(0)).cumsum()
 gid += s1.groupby(gid, group_keys=False).transform(lambda s: s.isna().any())
 s2 = pd.to_datetime(pd.Series(date_strs), format='%Y-%m-%d %H:%M:%S %f', errors='coerce').dt.floor('s')
 s2 = s2.groupby(gid, group_keys=False)\
        .apply(lambda s:
-           pd.date_range(
+           s1[s.index] if (idx := s.first_valid_index()) is None
-               s[s.first_valid_index()] - (s.first_valid_index() - s.index[0]) * pd.Timedelta('3s'),
+           else pd.date_range(s[idx] - (idx - s.index[0]) * pd.Timedelta('3s'),
-               freq='3s', periods=len(s)).to_series().set_axis(s.index))
+                              freq='3s', periods=len(s)).to_series().set_axis(s.index))
 print(s2)