python - 如何存储从 value_counts() 函数返回的信息

Question

我有以下数据框:

      import pandas as pd
      import numpy as np

      df_Station_Weather = pd.DataFrame({'ID': [102, 102, 558, 115, 115, 501, 501, 10, 10, 10, 10], 
                                         'Code_Instrument': ['SEN_wider1898', 'SEN_UV', 
                                                             'SEN_wider1898', 'SEN_wider1898', 
                                                             'SEN_rain1015', 'SEN_01', 'SEN_01', 
                                                             'SEN_AD', 'SEN_AD', 'SEN_AD', 
                                                             'SEN_AD']})

      print(df_Station_Weather)

      ID    Code_Instrument
      102   SEN_wider1898
      102   SEN_UV
      558   SEN_wider1898
      115   SEN_wider1898
      115   SEN_rain1015
      501   SEN_01
      501   SEN_01
      10    SEN_AD
      10    SEN_AD
      10    SEN_AD
      10    SEN_AD

我想计算特定仪器的数量。所以，我做了以下事情:

     list_Instrument = df_Station_Weather['Code_Instrument'].value_counts()

我只想选择计数的后三个。所以，我做了以下事情:

      list_Instrument_2 = list_Instrument.head(3)

我需要创建一个数组，其中包含三个最大的 ID 的名称。正是在这一部分，我有疑问。

我尝试使用以下代码构建数组:

      array = np.array(list_Instrument_2)

但是，这个创建的数组存储计数值，但我希望它存储 Code_Instrument 的名称。

      #Output:
      print(array)
      > array([4, 3, 2], dtype=int64)

     # Desired output
     array([SEN_AD, SEN_wider1898, SEN_01]

谢谢你。

Answer 1

尝试这个，

list_Instrument.head(3).index.values

array(['SEN_AD', 'SEN_wider1898', 'SEN_01'], dtype=object)