SQL 20251006

import%20marimo%0A%0A__generated_with%20%3D%20%220.17.6%22%0Aapp%20%3D%20marimo.App()%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_()%3A%0A%20%20%20%20import%20marimo%20as%20mo%0A%20%20%20%20return%20(mo%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20nav_menu%20%3D%20mo.nav_menu(%0A%20%20%20%20%20%20%20%20%7B%0A%20%20%20%20%20%20%20%20%20%20%20%20%22%2Findex.html%22%3A%20f%22%7Bmo.icon('lucide%3Ahome')%7D%20Home%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%22%2Fnotebooks%2FSQL_20251003.html%22%3A%20f%22%7Bmo.icon('lucide%3Aarrow-big-left')%7D%20Last%20Day%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%22%2Fnotebooks%2FSQL_20251007.html%22%3A%20f%22%7Bmo.icon('lucide%3Aarrow-big-right')%7D%20Next%20Day%22%2C%0A%20%20%20%20%20%20%20%20%7D%2C%0A%20%20%20%20%20%20%20%20orientation%3D%22horizontal%22%2C%0A%20%20%20%20)%0A%20%20%20%20nav_menu%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%20LeetCode%202854%0A%0A%20%20%20%20Table%3A%20Steps%0A%0A%20%20%20%20%7C%20Column%20Name%20%7C%20Type%20%7C%0A%20%20%20%20%7C-------------%7C------%7C%0A%20%20%20%20%7C%20user_id%20%20%20%20%20%7C%20int%20%20%7C%0A%20%20%20%20%7C%20steps_count%20%7C%20int%20%20%7C%0A%20%20%20%20%7C%20steps_date%20%20%7C%20date%20%7C%0A%0A%20%20%20%20(user_id%2C%20steps_date)%20is%20the%20primary%20key%20for%20this%20table.%20Each%20row%20of%20this%20table%20contains%20user_id%2C%20steps_count%2C%20and%20steps_date.%0A%0A%20%20%20%20Write%20a%20solution%20to%20calculate%203-day%20rolling%20averages%20of%20steps%20for%20each%20user.%0A%0A%20%20%20%20We%20calculate%20the%20n-day%20rolling%20average%20this%20way%3A%20For%20each%20day%2C%20we%20calculate%20the%20average%20of%20n%20consecutive%20days%20of%20step%20counts%20ending%20on%20that%20day%20if%20available%2C%20otherwise%2C%20n-day%20rolling%20average%20is%20not%20defined%20for%20it.%0A%0A%20%20%20%20Output%20the%20user_id%2C%20steps_date%2C%20and%20rolling%20average.%20Round%20the%20rolling%20average%20to%20two%20decimal%20places.%0A%0A%20%20%20%20Return%20the%20result%20table%20ordered%20by%20user_id%2C%20steps_date%20in%20ascending%20order.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_()%3A%0A%20%20%20%20%22%22%22%0A%20%20%20%20WITH%20full_rolling_info%20AS%20(%0A%20%20%20%20%20%20%20%20SELECT%0A%20%20%20%20%20%20%20%20%20%20%20%20user_id%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20steps_date%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20ROUND(AVG(steps_count)%20OVER%20(PARTITION%20BY%20user_id%20ORDER%20BY%20steps_date%20RANGE%20BETWEEN%20INTERVAL%20'2%20days'%20PRECEDING%20AND%20CURRENT%20ROW)%2C%202)%20AS%20rolling_average%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20COUNT(1)%20OVER%20(PARTITION%20BY%20user_id%20ORDER%20BY%20steps_date%20RANGE%20BETWEEN%20INTERVAL%20'2%20days'%20PRECEDING%20AND%20CURRENT%20ROW)%20AS%20rolling_count%0A%20%20%20%20%20%20%20%20FROM%0A%20%20%20%20%20%20%20%20%20%20%20%20Steps%0A%20%20%20%20)%0A%20%20%20%20SELECT%0A%20%20%20%20%20%20%20%20user_id%2C%0A%20%20%20%20%20%20%20%20steps_date%2C%0A%20%20%20%20%20%20%20%20rolling_average%0A%20%20%20%20FROM%0A%20%20%20%20%20%20%20%20full_rolling_info%0A%20%20%20%20WHERE%0A%20%20%20%20%20%20%20%20rolling_count%20%3D%203%0A%20%20%20%20ORDER%20BY%0A%20%20%20%20%20%20%20%201%2C%202%3B%0A%20%20%20%20%22%22%22%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_()%3A%0A%20%20%20%20import%20pandas%20as%20pd%0A%0A%20%20%20%20def%20rolling_average(steps%3A%20pd.DataFrame)%20-%3E%20pd.DataFrame%3A%0A%20%20%20%20%20%20%20%20df%20%3D%20steps.sort_values(%5B%22user_id%22%2C%20%22steps_date%22%5D).set_index(%22steps_date%22)%0A%0A%20%20%20%20%20%20%20%20df%5B%22rolling_average%22%5D%20%3D%20df.groupby(%22user_id%22)%5B%22steps_count%22%5D.rolling(%222D%22%2C%20closed%3D%22both%22).mean().round(2).reset_index(level%3D0%2C%20drop%3DTrue)%0A%20%20%20%20%20%20%20%20df%5B%22rolling_count%22%5D%20%3D%20df.groupby(%22user_id%22)%5B%22steps_count%22%5D.rolling(%222D%22%2C%20closed%3D%22both%22).count().reset_index(level%3D0%2C%20drop%3DTrue)%0A%20%20%20%20%20%20%20%20df.reset_index(inplace%3DTrue)%0A%20%20%20%20%20%20%20%20df%20%3D%20df%5Bdf%5B%22rolling_count%22%5D%20%3D%3D%203%5D%5B%5B%22user_id%22%2C%20%22steps_date%22%2C%20%22rolling_average%22%5D%5D.sort_values(%5B%22user_id%22%2C%20%22steps_date%22%5D)%0A%20%20%20%20%20%20%20%20return%20df%0A%20%20%20%20return%0A%0A%0Aif%20__name__%20%3D%3D%20%22__main__%22%3A%0A%20%20%20%20app.run()%0A