首页 > 解决方案 > 根据行索引创建新列

问题描述

我有一个类似于以下的数据框,

====================== DATAFRAME ===========================

     Question Value 
0    A          1              
1    B          4                
2    C          3                   
3    D          7               
4    E          45  
5    F          78          
6    G          44      
7    H          3           
8    I          4   
9    J          6       

我想根据索引号创建一个问题维度列,例如对于索引 = 0 到 3,问题维度 ='AA',对于索引 = 4 到 5,问题维度 ='BB'。其余的是“CC”。

====================== DATAFRAME ===========================

     Question Value  Question dimension
0    A          1           AA   
1    B          4           AA   
2    C          3           AA      
3    D          7           AA  
4    E          45          BB
5    F          78          BB
6    G          44          CC
7    H          3           CC
8    I          4           CC
9    J          6           CC

我怎样才能做到这一点?预先感谢您的帮助!!

标签: pythonindexing

解决方案


利用loc

df.loc[0:3, 'Question dimension'] = 'AA'
df.loc[4:5, 'Question dimension'] = 'BB'
df.loc[6:, 'Question dimension'] = 'CC'
     Question Value  Question dimension
0    A          1           AA   
1    B          4           AA   
2    C          3           AA      
3    D          7           AA  
4    E          45          BB
5    F          78          BB
6    G          44          CC
7    H          3           CC
8    I          4           CC
9    J          6           CC

推荐阅读