1. 程式人生 > 其它 >獲取物件屬性的集中方法

獲取物件屬性的集中方法

當我們拿到一個物件的引用時,如何知道這個物件是什麼型別、有哪些方法呢?

使用type()

首先,我們來判斷物件型別,使用type()函式:

基本型別都可以用type()判斷:

>>> type(123)
<class 'int'>
>>> type('str')
<class 'str'>
>>> type(None)
<type(None) 'NoneType'>

如果一個變數指向函式或者類,也可以用type()判斷:

>>> type(abs)
<class 'builtin_function_or_method'>
>>> type(a)
<class '__main__.Animal'>

但是type()函式返回的是什麼型別呢?它返回對應的Class型別。如果我們要在if語句中判斷,就需要比較兩個變數的type型別是否相同:

>>> type(123)==type(456)
True
>>> type(123)==int
True
>>> type('abc')==type('123')
True
>>> type('abc')==str
True
>>> type('abc')==type(123)
False

判斷基本資料型別可以直接寫intstr等,但如果要判斷一個物件是否是函式怎麼辦?可以使用types

模組中定義的常量:

>>> import types
>>> def fn():
...     pass
...
>>> type(fn)==types.FunctionType
True
>>> type(abs)==types.BuiltinFunctionType
True
>>> type(lambda x: x)==types.LambdaType
True
>>> type((x for x in range(10)))==types.GeneratorType
True

使用isinstance()

對於class的繼承關係來說,使用type()就很不方便。我們要判斷class的型別,可以使用isinstance()函式。

我們回顧上次的例子,如果繼承關係是:

object -> Animal -> Dog -> Husky

那麼,isinstance()就可以告訴我們,一個物件是否是某種型別。先建立3種類型的物件:

>>> a = Animal()
>>> d = Dog()
>>> h = Husky()

然後,判斷:

>>> isinstance(h, Husky)
True

沒有問題,因為h變數指向的就是Husky物件。

再判斷:

>>> isinstance(h, Dog)
True

h雖然自身是Husky型別,但由於Husky是從Dog繼承下來的,所以,h也還是Dog型別。換句話說,isinstance()判斷的是一個物件是否是該型別本身,或者位於該型別的父繼承鏈上。

因此,我們可以確信,h還是Animal型別:

>>> isinstance(h, Animal)
True

同理,實際型別是Dog的d也是Animal型別:

>>> isinstance(d, Dog) and isinstance(d, Animal)
True

但是,d不是Husky型別:

>>> isinstance(d, Husky)
False

能用type()判斷的基本型別也可以用isinstance()判斷:

>>> isinstance('a', str)
True
>>> isinstance(123, int)
True
>>> isinstance(b'a', bytes)
True

並且還可以判斷一個變數是否是某些型別中的一種,比如下面的程式碼就可以判斷是否是list或者tuple:

>>> isinstance([1, 2, 3], (list, tuple))
True
>>> isinstance((1, 2, 3), (list, tuple))
True
 總是優先使用isinstance()判斷型別,可以將指定型別及其子類“一網打盡”。

使用dir()

如果要獲得一個物件的所有屬性和方法,可以使用dir()函式,它返回一個包含字串的list,比如,獲得一個str物件的所有屬性和方法:

>>> dir('ABC')
['__add__', '__class__',..., '__subclasshook__', 'capitalize', 'casefold',..., 'zfill']

類似__xxx__的屬性和方法在Python中都是有特殊用途的,比如__len__方法返回長度。在Python中,如果你呼叫len()函式試圖獲取一個物件的長度,實際上,在len()函式內部,它自動去呼叫該物件的__len__()方法,所以,下面的程式碼是等價的:

>>> len('ABC')
3
>>> 'ABC'.__len__()
3

我們自己寫的類,如果也想用len(myObj)的話,就自己寫一個__len__()方法:

>>> class MyDog(object):
...     def __len__(self):
...         return 100
...
>>> dog = MyDog()
>>> len(dog)
100

剩下的都是普通屬性或方法,比如lower()返回小寫的字串:

>>> 'ABC'.lower()
'abc'

僅僅把屬性和方法列出來是不夠的,配合getattr()setattr()以及hasattr(),我們可以直接操作一個物件的狀態:

>>> class MyObject(object):
...     def __init__(self):
...         self.x = 9
...     def power(self):
...         return self.x * self.x
...
>>> obj = MyObject()

緊接著,可以測試該物件的屬性:

>>> hasattr(obj, 'x') # 有屬性'x'嗎?
True
>>> obj.x
9
>>> hasattr(obj, 'y') # 有屬性'y'嗎?
False
>>> setattr(obj, 'y', 19) # 設定一個屬性'y'
>>> hasattr(obj, 'y') # 有屬性'y'嗎?
True
>>> getattr(obj, 'y') # 獲取屬性'y'
19
>>> obj.y # 獲取屬性'y'
19

如果試圖獲取不存在的屬性,會丟擲AttributeError的錯誤:

>>> getattr(obj, 'z') # 獲取屬性'z'
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: 'MyObject' object has no attribute 'z'

可以傳入一個default引數,如果屬性不存在,就返回預設值:

>>> getattr(obj, 'z', 404) # 獲取屬性'z',如果不存在,返回預設值404
404

也可以獲得物件的方法:

>>> hasattr(obj, 'power') # 有屬性'power'嗎?
True
>>> getattr(obj, 'power') # 獲取屬性'power'
<bound method MyObject.power of <__main__.MyObject object at 0x10077a6a0>>
>>> fn = getattr(obj, 'power') # 獲取屬性'power'並賦值到變數fn
>>> fn # fn指向obj.power
<bound method MyObject.power of <__main__.MyObject object at 0x10077a6a0>>
>>> fn() # 呼叫fn()與呼叫obj.power()是一樣的
81