如何基于DataFrame中前几行的值标记某个值？

Question

数据如下所示：

Ctgr
A
B
B
C
D
E
F

我想要的结果是：

Ctgr|subctgr
  A |  A
  B |  A
  B |  A
  C |  A
  D |  D
  E |  D
  F |  D

也就是说，subctgr由每个“组”的第一行的值决定。
但是，我不知道如何区分python中的每个“组”，尽管它们由NaN值分隔

Answer 1

首先通过^{}测试缺失的行，然后通过与mask的^{}ed值比较获得groups的第一个值，并通过ffill创建由先前值填充的新列。上次按^{}设置新列：

#for test missing values
m = df['Ctgr'].isna()
#for test emtsy strings
#m = df['Ctgr'].eq('')
df['subctgr'] = np.where(m,np.nan,df['Ctgr'].where(m.ne(m.shift())).ffill())
print (df)
  Ctgr subctgr
0    A       A
1    B       A
2    B       A
3    C       A
4  NaN     NaN
5    D       D
6    E       D
7    F       D

详细信息：

print (df.assign(m = df['Ctgr'].isna(),
                 mask = m.ne(m.shift()),
                 first = df['Ctgr'].where(m.ne(m.shift())),
                 ffill = df['Ctgr'].where(m.ne(m.shift())).ffill(),
                 subctgr = np.where(m,np.nan,df['Ctgr'].where(m.ne(m.shift())).ffill())))
  Ctgr      m   mask first ffill subctgr
0    A  False   True     A     A       A
1    B  False  False   NaN     A       A
2    B  False  False   NaN     A       A
3    C  False  False   NaN     A       A
4  NaN   True   True   NaN     A     NaN
5    D  False   True     D     D       D
6    E  False  False   NaN     D       D
7    F  False  False   NaN     D       D

Answer 2

这可以使用pandas apply方法来完成，我假设如果字符是<='C'，那么您要放入'A'，否则'D'

假设数据帧的名称是df：

df['subctgr'] = df['Ctgr'].apply(lambda x: 'A' if x<='C' else 'D')

输出如下：

  Ctgr subctgr
0    A       A
1    B       A
2    B       A
3    C       A
4    D       D
5    E       D
6    F       D

Answer 3

这里有一个方法：

# create a group counter column
df['Counter'] = df['Ctgr'].isna().cumsum()

# drop NA
df2 = df[df['Ctgr'].notna()].reset_index(drop=True)

# join cols
def solve(f):
    # filter data
    f['col'] = f['Ctgr'].apply(lambda x: x + '|' + f['Ctgr'].iloc[0])
    return f

df2 = df2.groupby('Counter').apply(solve)

  Ctgr  Counter  col
0    A        0  A|A
1    B        0  B|A
2    B        0  B|A
3    C        0  C|A
4    D        1  D|D
5    E        1  E|D
6    F        1  F|D