code: def idxmax(self, axis=0, skipna=True, *args, **kwargs):
          
          if not all(d != np.dtype("O") for d in self._get_dtypes()):
              raise TypeError("reduction operation 'argmax' not allowed for this dtype")
          axis = self._get_axis_number(axis)
          return self._reduce_dimension(
              self._query_compiler.idxmax(axis=axis, skipna=skipna)
          )
docs: Get the index of the first occurrence of the max value of the axis.
  
          Args:
              axis (int): Identify the max over the rows (1) or columns (0).
              skipna (bool): Whether or not to skip NA values.
  
          Returns:
              A Series with the index for each maximum value for the axis
                  specified.