调用 class 中另一个方法的返回值的方法

Question

所以我想从同一个 class 的另一个方法调用返回值的方法。

class A():

    def __init__(self,data):
        self.data = data

    def values(self,key):
        return list(list(x.values())[0] for x in self.data['data'] if key in x.keys())

    def length(self):
        return len(self)

data ={"data":[{"country":"india"},{"state":"punjab"},{"country":"usa"}]}
obj = A(data)
res = obj.values('country') # returns ['india', 'usa']
res1 = obj.values('country').length() #  returns AttributeError: 'list' object has no attribute 'length'
print(res)
print(res1)

我希望 res 和 res1 都能工作。

我也尝试过使用@property 装饰器

class B():
    def __init__(self,data):
        self.data = data

    def __call__(self, key):
        self.key = key
        return self

    @property
    def values(self):
        self.valu =  list(list(x.values())[0] for x in self.data['data'] if self.key in x.keys())
        return self

    def length(self):
        return len(self.valu)

data ={"data":[{"country":"india"},{"state":"punjab"},{"country":"usa"}]}
obj = B(data)
res = obj('country').values.length()     #  returns 2
res1 = obj('country').values      # returns <__main__.A object at 0x103a9fbe0>
print(res)
print(res1)

这种方式 res 工作，但 res1 没有。

谢谢

Answer 1

.length()不是 python 列表 function。 改用len(listobject) 。

例如：

len(obj('country').values)

如果要打印值。 在您的第二个示例中：

#Use 
res1 = obj('country').data

#instead of
res1 = obj('country').values

使用以下代码获取国家/地区列表

res1 = [_dict.get('country') for _dict in obj('country').data.get('data') if 'country' in _dict.keys()]

print(res1)
#['india', 'usa']

最终更新的工作代码：

class B():
    def __init__(self,data):
        self.data = data

    def __call__(self, key):
        self.key = key
        return self

    @property
    def values(self):
        self.valu =  list(list(x.values())[0] for x in self.data['data'] if self.key in x.keys())
        return self

    def length(self):
        return len(self.valu)

data ={"data":[{"country":"india"},{"state":"punjab"},{"country":"usa"}]}
obj = B(data)
res = obj('country').values.length()     #  returns 2
res1 = [_dict.get('country') for _dict in obj('country').data.get('data') if 'country' in _dict.keys()]      # returns ['usa', 'india']
print(res)
print(res1)

Answer 2

我认为这里的困惑在于 object 是调用的.length方法。 在第一种情况下，它是在没有.length方法的list上调用的。 在第二种情况下，它在确实具有.length方法的B object 上调用。 最直接和首选的解决方案是@Subhrajyoti Das 建议的。

如果您仍然希望您的代码（如问题中所述）工作（或只是想知道这是否可以实际完成），您可以定义一个自定义列表 object 如下所示：

class CustomList(list):
    def length(self):
        return len(self)

现在，您将创建一个CustomList ，而不是制作一个list 。

编辑：按照评论中的要求添加完整的示例。

class CustomList(list):
    def length(self):
        return len(self)

class A():

    def __init__(self,data):
        self.data = data

    def values(self,key):
        return CustomList(CustomList(x.values())[0] for x in self.data['data'] if key in x.keys())

    def length(self):
        return len(self)

data ={"data":[{"country":"india"},{"state":"punjab"},{"country":"usa"}]}
obj = A(data)
res = obj.values('country') # returns ['india', 'usa']
res1 = obj.values('country').length() # 2
object has no attribute 'length'
print(res)
print(res1)