子选择多索引pandas数据帧以创建多个子集(使用字典)

时间:2017-06-23 15:29:53

标签: python pandas multi-index

我有一个类似于以下内容的数据集:

df_lenght = 240
df = pd.DataFrame(np.random.randn(df_lenght,2), columns=['a','b'] )
df['datetime'] = pd.date_range('23/06/2017', periods=df_lenght, freq='H')
unique_jobs = ['job1','job2','job3',]
job_id = [unique_jobs for i in range (1, int((df_lenght/len(unique_jobs))+1) ,1) ]
df['job_id'] = sorted( [val for sublist in job_id for val in sublist] )
df.set_index(['job_id','datetime'], append=True, inplace=True)

print(df[:5])返回:

                                     a         b
  job_id datetime                               
0 job1   2017-06-23 00:00:00 -0.067011 -0.516382
1 job1   2017-06-23 01:00:00 -0.174199  0.068693
2 job1   2017-06-23 02:00:00 -1.227568 -0.103878
3 job1   2017-06-23 03:00:00 -0.847565 -0.345161
4 job1   2017-06-23 04:00:00  0.028852  3.111738

如何为dataframes的每个值创建多个job_id,一个?那些被送入字典的人可以轻松找回吗? 感谢

1 个答案:

答案 0 :(得分:1)

您可以将groupby对象解压缩到字典中:

dfs = {job: df for job, df in df.groupby(level='job_id')}
相关问题