用Python循环提取每个月的数据

Question

整个dataset一共12个月的数据，如何用for month in range 循环提取每个月的数据？

Answer 1

假如是如2020-01-01这样日期数据，可使用pandas提取。
样例数据：

date,open,close,high,low,volume
2021-01-04,4.31,4.42,4.45,4.31,84006.0
2021-01-05,4.45,4.3,4.46,4.26,93427.0
2021-01-06,4.29,4.19,4.3,4.14,91958.0
2021-01-07,4.19,3.98,4.19,3.96,95370.0
2021-02-01,3.95,4.04,4.05,3.93,54993.0
2021-02-02,4.13,4.04,4.13,4.02,45825.0
2021-02-03,4.04,4.03,4.16,3.97,63093.0
2021-02-04,4.0,3.98,4.04,3.93,52044.0
2021-03-01,4.9,5.14,5.28,4.86,258844.0
2021-03-02,5.07,4.78,5.09,4.74,191115.0
2021-03-03,4.78,4.86,4.97,4.68,141404.0
2021-03-04,4.82,4.75,4.9,4.73,93465.0

参考代码：

import pandas as pd
df=pd.read_csv('data/600889_1.csv')
for i in range(1,13):
    x=str(i).zfill(2)
    if any(df['date'].str[5:7]==x):
        df0=df[df['date'].str[5:7]==x]
        print(df0)
        df0.to_csv(f'data/600889_1_{x}.csv',index=False)

如有帮助，请点采纳。