python - pandas Dataframe.loc 如何接受 [...] 语法？

Question

我已阅读此文档:

https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.loc.html

您可以使用类似 df.loc[df['shield'] > 6, ['max_speed']] 的语法.

我尝试使用 Github 并发现:

假设您有 pandas.core.frame.DataFrame对象，即 DataFrame叫df .

df.loc的类型是 pandas.core.indexing._LocIndexer .

尽管如此，我还是无法解决这些问题:

1. 如何使 Python 函数/类接受上述语法？

2. pandas.core.frame.DataFrame的源代码在哪里是属性(property) self.loc定义？？

Answer 1

1. 如何让类接受一般语法是通过实现 __getitem__这是运算符重载的一个例子。这允许使用 [] 对该类的对象进行索引。例如:

   class get_item_example(object):
    def __getitem__(self, key):
            print(key)
   

   尝试一下:

   >>> gi = get_item_example()
   >>> gi['a']
   a
   >>> gi[['a','b','c']]
   ['a', 'b', 'c']
   >>> gi['a','b','c']
   ('a', 'b', 'c')
   

   在 df.loc[df['shield'] > 6, ['max_speed']] 的情况下，传递给 __getitem__ 的 key 是包含 df['shield'] > 6 返回的 pandas 系列和单个项目列表 ['max_speed'] 的元组。

2. 在pandas源代码中，pandas.core.indexing._LocIndexer从pandas.core.indexing继承了__getitem__的实现。 _LocationIndexer。实现在这里:https://github.com/pandas-dev/pandas/blob/61362be9ea4d69b33ae421f1f98b8db50be611a2/pandas/core/indexing.py#L1374