python - Pandas:如何在分组行中添加行数

Question

所以我有数据框:

>>> df2
      text
0  0  a  
0  1  b
0  2  c
0  3  d
1  4  e
1  5  f  
1  6  g  
2  7  h  
2  8  1

如何创建另一列，其中包含 level=0 索引中每行的计数器？

我尝试了以下代码(我需要获取 df['counter'] 列):

current_index = ''
for index, row in df.iterrows():
  if index[0] != current_index:
    current_index = index[0]
    df[(df.index == current_index)]['counter'] = np.arange(len(df[(df.index == current_index)].index))

以及以下代码:

df2 = pd.DataFrame()
for group, df in df1.groupby('level_0_column'):
  df0 = df0.sort_values(by=['level_1_column'])
  df['counter'] = list(df.reset_index().index.values + 1)
  df2 = df2.append(df0)

我的 DataFrame 中有大约 650K 行...进入无限循环。请指教

Answer 1

我相信您正在沿着第 0^th 列索引 + cumcount 查找 groupby:

df['counter'] = df.groupby(level=0).cumcount() + 1
df

    text  counter
0 0    a        1
  1    b        2
  2    c        3
  3    d        4
1 4    e        1
  5    f        2
  6    g        3
2 7    h        1
  8    1        2