pandas_从股本变动数据中获取季度末股本数

目录

1. 股本变动数据

2. 代码-读取股本变动数据

3. 代码-按照secID进行分组，并按季度重采样

4. 代码-去除无用的changeDate普通列 

5. 数据

---

1. 股本变动数据

股本变动数据记录了每次股本变动的日期

2. 代码-读取股本变动数据

import pandas as pd
import numpy as np
df = pd.read_csv('shares.csv',encoding='utf-8')
column_list = ['secID','secShortName','changeDate','totalShares']
df = df.loc[:,column_list].copy()
df.head()

3. 代码-按照secID进行分组，并按季度重采样

# 对secID进行分组
df_group = df.groupby('secID')
res_df = pd.DataFrame()
for name,group in df_group:
    group['changeDate'] = pd.to_datetime(group['changeDate'])
    one_df = group.resample('Q',on='changeDate').last()
    one_df00 = one_df.fillna(method='ffill')
    res_df = pd.concat([res_df,one_df00])
res_df.to_csv('111.csv',encoding='utf-8')
res_df.head()

 

4. 代码-去除无用的changeDate普通列 

temp_df = res_df.drop('changeDate',axis=1)
temp_df.head()

5. 数据

 链接：https://pan.baidu.com/s/1SpOSCyA300punggvk5SNNQ 
提取码：y8ic