Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
893 views
in Technique[技术] by (71.8m points)

python - How to split dataframe in pandas

I have dataframe below

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

I would like to split dataframe by df.A

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

and

A B C
1 d k
1 e l

and

A B C
2 f m
2 g n

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

See Question&Answers more detail:os

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Reply

0 votes
by (71.8m points)

You can create dictionary of DataFrames by dict comprehension:

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

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
OGeek|极客中国-欢迎来到极客的世界,一个免费开放的程序员编程交流平台!开放,进步,分享!让技术改变生活,让极客改变未来! Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

...