在python數據框中創建具有多個名稱的列

Question

我有一個像這樣的數據框df：

        Country     Value   
          US         300
          FR         360
          IT         500
          NL         450
          CH         700
          CN         233
          NF         123
          AR         456
          BR         129
          NG         423
          ..          .. 
          ..          ..
          GR         560
          CY         145
          ES         890
          PR         783
          BL         123

我知道df中的這些國家是每種“實驗類型”的前5個國家。 實驗類型是我創建的列表，它是： my_list = [1, 2, 3, 4, 5, ......,20 ] 。 因此，前5個國家屬於實驗1，接下來的5個國家屬於實驗2，……...，后5個國家屬於實驗20。

我想創建一列，該列將通過實驗指示國家/地區，因此我想要這樣做：

    Experiment       Country     Value   
       1               US         300
                       FR         360
                       IT         500
                       NL         450
                       CH         700
       2               CN         233
                       NF         123
                       AR         456
                       BR         129
                       NG         423
                       ..          .. 
                       ..          ..
      20               GR         560
                       CY         145
                       ES         890
                       PR         783
                       BL         123

我知道在使用groupby函數創建數據框時可以執行類似的操作，但是在創建初始數據框后，您能幫上什么是最好的方法嗎？

Answer 1

我相信最好的方法是使用實​​驗值添加一個新列，如下所示：

df['Experiment'] = np.arange(len(df)) // 5
print (df)
   Country  Value  Experiment
0       US    300           0
1       FR    360           0
2       IT    500           0
3       NL    450           0
4       CH    700           0
5       CN    233           1
6       NF    123           1
7       AR    456           1
8       BR    129           1
9       NG    423           1
10      GR    560           2
11      CY    145           2
12      ES    890           2
13      PR    783           2
14      BL    123           2