根据行索引创建一个新列

时间:2020-02-19 01:16:37

标签: python indexing

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

======================数据框======================== ===

     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”。

======================数据框======================== ===

     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

我该如何实现?预先感谢您的帮助!

1 个答案:

答案 0 :(得分:0)

使用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