Python 中检查对象属性是否分配了数据帧的最有效方法？

Question

我正在尝试构建 class 中所有属性名称的列表，这些属性是 pandas DataFrames 或 Series。

以下工作但似乎很慢，因为它遍历了 dir() 列出的每个属性，包括特殊方法（特殊方法不是数据帧，因此无需检查它们）。它似乎也在其他方法中执行了 print 语句，因此不确定其他属性是否正在运行（不需要）。

my_instance = MyClass()
pd_objects = [attr for attr in dir(my_instance) if 
              isinstance(getattr(my_instance, attr), pd.DataFrame)
              or isinstance(getattr(my_instance, attr), pd.Series)]

关于快速列出分配给 pandas 对象的属性的更有效方法的任何想法，也许使用 inspect 模块？

Answer 1

我不知道使用 inspect() 模块的任何方法，但是除了 dir() 之外，您可以尝试使用 vars() ，它 returns __dict__ 用于传递给它的对象，因此它不会有特殊方法等。示例 -

my_instance = MyClass()
pd_objects = [attr for attr in vars(my_instance) if 
          isinstance(getattr(my_instance, attr), pd.DataFrame)
          or isinstance(getattr(my_instance, attr), pd.Series)]

请注意，这仍然会评估属性，因为如果不评估属性，就无法确定它是什么 contains/returns（这很可能是您看到正在执行的 print 语句).

vars([object])-

Return the __dict__ attribute for a module, class, instance, or any other object with a __dict__ attribute.

Answer 2

您想使用__dict__。

class A(object):
    def __init__(self):
        self.df2 = pd.DataFrame({'df2': range(5)})
        self.sr2 = pd.Series(range(5))
        self.x = 4
        self.y = 'abc'
        self.z = []


def get_attributes_of_type(cls, types):
    return [i for _, i in cls.__dict__.iteritems() if isinstance(i, types)]

然后像这样使用它：

my_A = A()
get_attributes_of_type(my_A, (pd.Series, pd.DataFrame))
[0    0
 1    1
 2    2
 3    3
 4    4
 dtype: int64,    df2
 0    0
 1    1
 2    2
 3    3
 4    4]

get_attributes_of_type(my_A, list)
[[]]

get_attributes_of_type(my_A, str)
['abc']

get_attributes_of_type(my_A, int)
[4]

Python 中检查对象属性是否分配了数据帧的最有效方法？

Most efficient way in Python to check if object attributes are assigned DataFrames?

python

attributes

inspect

dataframe

pandas