使用相同值按块分组 pandas dataframe

Question

我有一个大的 dataframe，其结构如下：
|'类型'| |'col2'| |'col3'|
| ----- | | -----| |-----|
'一个'
'乙'
'C'
'C'
'C'
'乙'
C
C
C
一个
乙
C
C
乙
C
一个

所以类型就像层次结构； 与具有一个或多个 C 的一个或多个 B 一样。 我想把这个 dataframe 分成两种不同的块：

• 从 A 到下一个 A 的 1 个块（所有 B 和每个 A 的 C）
• 每个 A 块中的 1 个块，从 B 到下一个 B（A 中每个 B 的所有 C）

我怎样才能做到这一点？

Answer 1

IIUC，您希望 col2 具有以 A 开头的组和以 B 开头的 col3 子组：

df['col2'] = df['Type'].eq('A').cumsum()
df['col3'] = df['Type'].eq('B').groupby(df['col2']).cumsum()

output：

   Type  col2  col3
0     A     1     0
1     B     1     1
2     C     1     1
3     C     1     1
4     C     1     1
5     B     1     2
6     C     1     2
7     C     1     2
8     C     1     2
9     A     2     0
10    B     2     1
11    C     2     1
12    C     2     1
13    B     2     2
14    C     2     2
15    A     3     0

然后，您可以使用 col2/col3 到groupby ：

m = df[['col2', 'col3']].ne(0).all(1)
for name, g in df[m].groupby(['col2', 'col3']):
    print(f'group {name}')
    print(g)

output：

group (1, 1)
  Type  col2  col3
1    B     1     1
2    C     1     1
3    C     1     1
4    C     1     1
group (1, 2)
  Type  col2  col3
5    B     1     2
6    C     1     2
7    C     1     2
8    C     1     2
group (2, 1)
   Type  col2  col3
10    B     2     1
11    C     2     1
12    C     2     1
group (2, 2)
   Type  col2  col3
13    B     2     2
14    C     2     2