Python 3.x 如何在python中对列中的字符串进行分组？

我有一个数据帧

   PROD  TYPE  QUANTI  
0  wood    i2      20     
1    tv    ut1      30    
2  tabl    il3      50    
3   rmt    z1      40     
4   zet    u1      60   
5    rm    t1      60      
6    rt    t2      80    
7   dud    i4      40  

我想将列“TYPE”分组到（I、u、z、y…等）的组类别中

预期产出

   PROD      TYPE    QUANTI  
0  wood    i_group      20     
1    tv    ut_group      30    
2  tabl    il_group      50    
3   rmt    z_group      40     
4   zet    y_group      60   
5    rm    t_group      60      
6    rt    t_group      80    
7   dud    i_group      40  

用于更换

\u组的编号

：

df['TYPE'] = df['TYPE'].replace('\d+', '_group', regex=True) 
print (df)
   PROD      TYPE  QUANTI
0  wood   i_group      20
1    tv  ut_group      30
2  tabl  il_group      50
3   rmt   z_group      40
4   zet   u_group      60
5    rm   t_group      60
6    rt   t_group      80
7   dud   i_group      40

如有可能，某些数值不带数字，请使用：

df['TYPE'] = df['TYPE'].replace('\d+', '', regex=True) + '_group'

---

df[“TYPE”]=df[“TYPE”].str[0]+“_group”

？您能检查一下编辑好的吗。。。喜欢它可以分组使用包含或任何简单的方式来做到这一点。。。因为.str[0]太具体了。就像我有一个双字母或三字母的列类型名称一样，这将是一个问题@耶斯雷尔