是否可以構建一個自動插值的熊貓系列？

Question

對於任何給定的索引，是否可以生成一個插值它的值的序列。我有一個預定義的插值方案，我希望規定，我寧願調用者不自己應用插值，以避免任何錯誤的可能性。

class InterpolatedSeries(pd.Series): 
    pass # magic? 

s = pd.Series([1, 3], index=[1, 3]) 
i = InterpolatedSeries(s, forward='nearest', backward='nearest', middle='linear') 

主叫方將收到i結果，他們現在可以要求任何價值，而且我確信他們得到了符合規定的插值方案的價值。插值當然不是可預先計算的（因爲我們不知道他們會提前請求哪些點）還是可緩存的（因爲我們不知道他們要求多少點），但重要的是沒有併發症爲來電者。

這可能嗎？

>>> i[[0, 0.11234, 1, 2, 2.367, 3, 4]] 
... pd.Series([1, 1, 1, 2, 2.367, 3, 3], index=[0, 0.11234, 1, 2, 2.367, 3, 4]) 

Answer 1

使用__getitem__。它被稱爲蟒魔術方法http://www.diveintopython3.net/special-method-names.html

class InterpolatedSeries(pd.Series): 
    def __init__(self, values, forward='nearest', backward='nearest', middle='linear'): 
     super().__init__(values) 
     self.forward = forward 
     self.backward = backward 
     self.middle = middle 

    def __getitem__(self, key): 
     # get the stored values 
     values = super().__getitem__(key) 
     # Do interpolation 
     return values 

或

class InterpolatedSeries(pd.Series): 
    def __init__(self, values, forward='nearest', backward='nearest', middle='linear'): 
     super().__init__(values) 
     self.forward = forward 
     self.backward = backward 
     self.middle = middle 

    def __setitem__(self, key, value): 
     # Do interpolation 
     super().__setitem__(key, value) 

另一種選擇是創建你自己的類，它使用的底層數據結構進行交互。這個類不會從pd.Series繼承，而是一個對象。

class InterpolatedSeries(object): 
    def __init__(self, values, forward='nearest', backward='nearest', middle='linear'): 
     self.data = values 
     self.forward = forward 
     self.backward = backward 
     self.middle = middle 

    def __getitem__(self, key): 
     values = self.data.__getitem__(key) 
     # Do interpolation 
     return values 

    def __getattribute__(self, key): # maybe __getattr__ if this doesn't work 
     """Return the stored pandas series item if the method or attribute was not found. This allows your to_csv method to work""" 
     try: 
      return super().__getattribute__(key) 
     except AttributeError: 
      pass 
     return self.data.__getattribute__(key) # Call the stored pandas series method if not found. 

    def __dir__(self): 
     """Return the list of attributes. (Most code autocomplete features use this, so this will find your pandas series methods for autocomplete in IDEs). """ 
     values = dir(self.data) 
     return values + super().__dir__() 

上面的可能不是最好的方法，但它確實通過使其更容易訪問後臺熊貓系列方法添加了一些靈活性。