根據行值在數據框中創建新列

Question

我希望能夠在不使用 for 循環或 df.apply() 的情況下獲得以下結果

每行的結果應該是直到組索引的行值。

   group  0  1  2  3  4  5  6  7
0      2  a  b  c  d  e  f  g  h
1      5  s  t  u  v  w  x  y  z
2      7  a  b  c  d  e  f  g  h

   group                    result
0      2                 [a, b, c]
1      5        [s, t, u, v, w, x]
2      7  [a, b, c, d, e, f, g, h]

Answer 1

在DataFrame.query和最后一個聚合list中使用DataFrame.melt 、過濾group列和變量列：

s = (df.melt('group', ignore_index=False)
       .astype({'variable':int})
       .query("group >= variable")
       .groupby(level=0)['value']
       .agg(list))

df = df[['group']].join(s.rename('result'))
print (df)
   group                    result
0      2                 [a, b, c]
1      5        [s, t, u, v, w, x]
2      7  [a, b, c, d, e, f, g, h]

或使用apply ：

df = (df.set_index('group')
        .rename(columns=int)
        .apply(lambda x: list(x[x.index <= x.name]), axis=1)
        .reset_index(name='result'))
print (df)
   group                    result
0      2                 [a, b, c]
1      5        [s, t, u, v, w, x]
2      7  [a, b, c, d, e, f, g, h]