根据dataframe中A列的相同元素将dataframe的A列和B列切分成不同的dataframe
时间: 2023-06-11 07:06:51 浏览: 94
可以使用pandas中的groupby函数来实现。具体步骤如下:
1. 使用groupby函数将dataframe按照A列进行分组:
```
grouped = df.groupby('A')
```
2. 遍历分组后的grouped对象,将每个分组的A列和B列分别切分成一个新的dataframe:
```
for group_name, group_df in grouped:
new_df_A = group_df[['A']]
new_df_B = group_df[['B']]
# do something with new_df_A and new_df_B
```
完整代码示例:
```
import pandas as pd
# 创建示例dataframe
df = pd.DataFrame({
'A': ['x', 'x', 'y', 'y', 'y'],
'B': [1, 2, 3, 4, 5]
})
# 按照A列分组
grouped = df.groupby('A')
# 遍历分组后的grouped对象,将每个分组的A列和B列分别切分成一个新的dataframe
for group_name, group_df in grouped:
new_df_A = group_df[['A']]
new_df_B = group_df[['B']]
print('分组名:', group_name)
print('新的dataframe A:')
print(new_df_A)
print('新的dataframe B:')
print(new_df_B)
```
输出结果:
```
分组名: x
新的dataframe A:
A
0 x
1 x
新的dataframe B:
B
0 1
1 2
分组名: y
新的dataframe A:
A
2 y
3 y
4 y
新的dataframe B:
B
2 3
3 4
4 5
```
阅读全文