Создать новый столбец на основе индекса строки

Question

У меня есть фрейм данных, подобный следующему,

====================== 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       

Я хочу создать столбец измерения вопроса на основе номер индекса , например, для index = От 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

Как мне этого добиться? Заранее благодарю за помощь !!

Answer 1

Использование 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