如何在 pandas 中拆分数据框 [英] How to split dataframe in pandas

问题描述

我下面有数据框

A B C
0 a h
0 b i
0 c j
1 d k
1 e l
2 f m
2 g n

我想按df.A分割数据帧

I would like to split dataframe by df.A

A B C
0 a h
0 b i 
0 c j

和

A B C
1 d k
1 e l

和

A B C
2 f m
2 g n

我尝试了groupby，但效果不佳.如何将数据框拆分为多个数据框?

I tried groupby but It didnt work well. how can I split dataframe to multiple dataframe?

推荐答案

您可以通过dict comprehension创建DataFrames的dictionary:

dfs = {k:v for k, v in df.groupby('A')}
print (dfs)

{0:    A  B  C
0  0  a  h
1  0  b  i
2  0  c  j, 1:    A  B  C
3  1  d  k
4  1  e  l, 2:    A  B  C
5  2  f  m
6  2  g  n}

print (dfs[0])
   A  B  C
0  0  a  h
1  0  b  i
2  0  c  j

print (dfs[1])
   A  B  C
3  1  d  k
4  1  e  l

如有必要，您可以重置索引:

If necessary you can reset index:

dfs = {k:v.reset_index(drop=True) for k, v in df.groupby('A')}
print (dfs)
{0:    A  B  C
0  0  a  h
1  0  b  i
2  0  c  j, 1:    A  B  C
0  1  d  k
1  1  e  l, 2:    A  B  C
0  2  f  m
1  2  g  n}

print (dfs[1])
   A  B  C
0  1  d  k
1  1  e  l

print (dfs[2])
   A  B  C
0  2  f  m
1  2  g  n

这篇关于如何在 pandas 中拆分数据框的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！