在自己的类中使用来自 pandas 的 iloc 和 loc

Question

我想创建一个自己的类ExampleClass，将数据存储在pandas DataFrame 中。我这样做是为了创建自己的方法来操作数据。

我的问题是： 是否可以使用ExampleClass 的对象，例如：

import pandas as pd

class ExampleClass:
    def __init__(self, data: pd.DataFrame):
        self.data = data
        self.other_value = 42

obj = ExampleClass(pd.DataFrame([1,2,3]))
part_of_obj = obj.iloc[:2]

type(part_of_obj) == ExampleClass 和 part_of_obj.data == obj.data.iloc[:2] 在哪里？

所以我想要一份obj 但带有obj.data.iloc[:2] 的副本。

我不想使用__getitem__，因为那样我就被绑定到 iloc 或 loc 并且不能同时使用两者。而且我不希望自己创建一个名为iloc() 的方法，因为我无法使用[] 轻松访问它。

也许我可以通过继承来实现，但我已经编写了我的类，它使用 DataFrame 来操作数据，这使得更改所有编写的内容变得非常困难。

Answer 1

您可以通过属性公开iloc，然后您也可以添加loc，以同样的方式对数据框进行子集化。

@property
def iloc(self):
  return self.data.iloc

要返回相同类型的对象，也许可以使用一个帮助类来包装对__getitem__的调用：

class ExampleClass:
  class Subsetter:
    def __init__(self, locator):
      self.locator = locator

    def __getitem__(self, *vargs, **kwargs):
      return ExampleClass(self.locator.__getitem__(*vargs, **kwargs))

  def __init__(self, data: pd.DataFrame):
      self.data = data
      self.other_value = 42

  @property
  def iloc(self):
    return self.Subsetter(self.data.iloc)

obj = ExampleClass(pd.DataFrame([1,2,3]))
part_of_obj = obj.iloc[:2]