Итерация над атрибутами объекта в python
У меня есть объект python с несколькими атрибутами и методами. Я хочу перебрать атрибуты объекта.
class my_python_obj(object):
attr1='a'
attr2='b'
attr3='c'
def method1(self, etc, etc):
#Statements
Я хочу сгенерировать словарь, содержащий все атрибуты объектов и их текущие значения, но я хочу сделать это динамически (так что если позже я добавлю еще один атрибут, мне не нужно помнить о том, чтобы обновлять мою функцию как а).
В php-переменные могут использоваться как ключи, но объекты в python не подлежат описанию, и если я использую точечную нотацию для этого, он создает новый атрибут с именем моего var, что не является моим намерением.
Просто, чтобы сделать все более ясным:
def to_dict(self):
'''this is what I already have'''
d={}
d["attr1"]= self.attr1
d["attr2"]= self.attr2
d["attr3"]= self.attr3
return d
·
def to_dict(self):
'''this is what I want to do'''
d={}
for v in my_python_obj.attributes:
d[v] = self.v
return d
Обновление:
С атрибутами я имею в виду только переменные этого объекта, а не методы.
Ответы
Ответ 1
Предполагая, что у вас есть класс, например
>>> class Cls(object):
... foo = 1
... bar = 'hello'
... def func(self):
... return 'call me'
...
>>> obj = Cls()
вызов dir
объекта возвращает все атрибуты этого объекта, включая специальные атрибуты python. Хотя некоторые атрибуты объекта вызываются, например методы.
>>> dir(obj)
['__class__', '__delattr__', '__dict__', '__doc__', '__format__', '__getattribute__', '__hash__', '__init__', '__module__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', '__weakref__', 'bar', 'foo', 'func']
Вы всегда можете отфильтровать специальные методы, используя понимание списка.
>>> [a for a in dir(obj) if not a.startswith('__')]
['bar', 'foo', 'func']
или если вы предпочитаете карту/фильтры.
>>> filter(lambda a: not a.startswith('__'), dir(obj))
['bar', 'foo', 'func']
Если вы хотите отфильтровать методы, вы можете использовать встроенный callable
как проверку.
>>> [a for a in dir(obj) if not a.startswith('__') and not callable(getattr(obj,a))]
['bar', 'foo']
Вы также можете проверить разницу между вашим классом и его родителем.
>>> set(dir(Cls)) - set(dir(object))
set(['__module__', 'bar', 'func', '__dict__', 'foo', '__weakref__'])
Ответ 2
обычно применяет метод __iter__
в вашем классе и выполняет итерацию по атрибутам объекта или помещает этот класс mixin в ваш класс.
class IterMixin(object):
def __iter__(self):
for attr, value in self.__dict__.iteritems():
yield attr, value
Ваш класс:
>>> class YourClass(IterMixin): pass
...
>>> yc = YourClass()
>>> yc.one = range(15)
>>> yc.two = 'test'
>>> dict(yc)
{'one': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14], 'two': 'test'}
Ответ 3
Объекты в python сохраняют свои атрибуты (включая функции) в dict, называемом __dict__
. Вы можете (но обычно не должны) использовать это для прямого доступа к атрибутам. Если вам нужен только список, вы также можете вызвать dir(obj)
, который возвращает итерабельность со всеми именами атрибутов, которые затем можно передать в getattr
.
Однако, нужно что-либо делать с именами переменных, как правило, плохой дизайн. Почему бы не сохранить их в коллекции?
class Foo(object):
def __init__(self, **values):
self.special_values = values
Затем вы можете перебирать клавиши с помощью for key in obj.special_values:
Ответ 4
class someclass:
x=1
y=2
z=3
def __init__(self):
self.current_idx = 0
self.items = ["x","y","z"]
def next(self):
if self.current_idx < len(self.items):
self.current_idx += 1
k = self.items[self.current_idx-1]
return (k,getattr(self,k))
else:
raise StopIteration
def __iter__(self):
return self
то просто назовите его как итерируемый
s=someclass()
for k,v in s:
print k,"=",v
Ответ 5
Правильный ответ на этот вопрос заключается в том, что вы не должны. Если вы хотите, чтобы этот тип вещей просто использовал dict, или вам нужно явно добавлять атрибуты в какой-либо контейнер. Вы можете автоматизировать это, узнав о декораторах.
В частности, кстати, метод method1 в вашем примере не хуже атрибута.