微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

Python中检查对象属性是否分配了DataFrame的最有效方法?

我试图建立一个类中所有属性名称列表,这些类要么是pandas DataFrames,要么是Series.

以下工作有效,但似乎很慢,因为它遍历了dir()列出的每个属性,包括特殊方法(这些特殊方法不会是DataFrames,因此不需要检查它们).它似乎也可以用其他方法执行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)] 

有什么想法可以更高效地快速列出分配给熊猫对象的属性,也许使用检查模块?

解决方法:

您要使用__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]

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐