Python中operator模塊的操作符使用示例總結(jié)

更新時(shí)間：2016年06月28日 17:11:04 作者：nummy

operator模塊中包含了Python的各種內(nèi)置操作符,諸如邏輯、比較、計(jì)算等,這里我們針對(duì)一些常用的操作符來作一個(gè)Python中operator模塊的操作符使用示例總結(jié):

operator模塊是python中內(nèi)置的操作符函數(shù)接口，它定義了一些算術(shù)和比較內(nèi)置操作的函數(shù)。operator模塊是用c實(shí)現(xiàn)的，所以執(zhí)行速度比python代碼快。

邏輯操作

from operator import *

a = [1, 2, 3]
b = a
print 'a =', a
print 'b =', b
print 

print 'not_(a)   :', not_(a)
print 'truth(a)   :', truth(a)
print 'is_(a, b)  :', is_(a, b)
print 'is_not(a, b) :', is_not(a, b)

打印結(jié)果：

a = [1, 2, 3]
b = [1, 2, 3]
not_(a)   : False
truth(a)  : True
is_(a, b)  : True
is_not(a, b): False

可以通過結(jié)果知道，operator的一些操作函數(shù)與原本的運(yùn)算是相同的。

比較操作符
operator提供豐富的比較操作。

a = 3
b = 5
print 'a =', a
print 'b =', b
print 

for func in (lt, le, eq, ne, ge, gt):
  print '{0}(a, b):'.format(func.__name__), func(a, b)

打印結(jié)果

a = 3
b = 5

lt(a, b): True
le(a, b): True
eq(a, b): False
ne(a, b): True
ge(a, b): False
gt(a, b): False

這些函數(shù)等價(jià)于<、<=、==、>=和>的表達(dá)式語法。

算術(shù)操作符
處理數(shù)字的算術(shù)操作符也得到支持。

a, b, c, d = -1, 2, -3, 4

print 'a =', a
print 'b =', b
print 'c =', c
print 'd =', d
 
print '\nPositive/Negative:'
print 'abs(a):', abs(a)
print 'neg(a):', neg(a)
print 'neg(b):', neg(b)
print 'pos(a):', pos(a)
print 'pos(b):', pos(b)

打印結(jié)果

a = -1
b = 2
c = -3
d = 4

Positive/Negative:
abs(a): 1
neg(a): 1
neg(b): -2
pos(a): -1
pos(b): 2

abs返回值得絕對(duì)值，neg返回(-obj), pos返回(+obj)。

a = -2
b = 5.0

print 'a =', a
print 'b =', b
 
print '\nArithmetic'
print 'add(a, b)    :', add(a, b)
print 'div(a, b)    :', div(a, b)
print 'floordiv(a, b)  :', floordiv(a, b)
print 'mod(a, b)    :', mod(a, b)
print 'mul(a, b)    :', mul(a, b)
print 'pow(a, b)    :', pow(a, b)
print 'sub(a, b)    :', sub(a, b)
print 'truediv(a, b)  :', truediv(a, b)

打印結(jié)果

a = -2
b = 5.0

Arithmetic
add(a, b)    : 3.0
div(a, b)    : -0.4
floordiv(a, b)  : -1.0
mod(a, b)    : 3.0 # 查看負(fù)數(shù)取模
mul(a, b)    : -10.0
pow(a, b)    : -32.0
sub(a, b)    : -7.0
truediv(a, b)  : -0.4

mod表示取模， mul 表示相乘，pow是次方, sub表示相減

a = 2
b = 6

print 'a =', a
print 'b =', b

print '\nBitwise:'
print 'and_(a, b)  :', and_(a, b)
print 'invert(a)  :', invert(a)
print 'lshift(a, b) :', lshift(a, b)
print 'or_(a, b)  :', or_(a, b)
print 'rshift(a, b) :', rshift(a, b)
print 'xor(a, b)  :', xor(a, b)

打印結(jié)果

a = 2
b = 6

Bitwise:
and_(a, b)  : 2
invert(a)  : -3
lshift(a, b) : 128
or_(a, b)  : 6
rshift(a, b) : 0
xor(a, b)  : 4

and 表示按位與, invert 表示取反操作, lshift表示左位移, or表示按位或, rshift表示右位移,xor表示按位異或。

原地操作符
即in-place操作， x += y 等同于 x = iadd(x, y), 如果復(fù)制給其他變量比如z = iadd(x, y)等同與z = x; z += y。

a = 3
b = 4
c = [1, 2]
d = ['a', 'b']

print 'a =', a
print 'b =', b
print 'c =', c
print 'd =', d
print

a = iadd(a, b)
print 'a = iadd(a, b) =>', a
print

c = iconcat(c, d)
print 'c = iconcat(c, d) =>', c

屬性和元素的獲取方法
operator模塊最特別的特性之一就是獲取方法的概念，獲取方法是運(yùn)行時(shí)構(gòu)造的一些可回調(diào)對(duì)象，用來獲取對(duì)象的屬性或序列的內(nèi)容，獲取方法在處理迭代器或生成器序列的時(shí)候特別有用，它們引入的開銷會(huì)大大降低lambda或Python函數(shù)的開銷。

from operator import *
class MyObj(object):
  def __init__(self, arg):
    super(MyObj, self).__init__()
    self.arg = arg
  def __repr__(self):
    return 'MyObj(%s)' % self.arg

objs = [MyObj(i) for i in xrange(5)]
print "Object:", objs

g = attrgetter("arg")
vals = [g(i) for i in objs]
print "arg values:", vals

objs.reverse()
print "reversed:", objs
print "sorted:", sorted(objs, key=g)

結(jié)果：

Object: [MyObj(0), MyObj(1), MyObj(2), MyObj(3), MyObj(4)]
arg values: [0, 1, 2, 3, 4]
reversed: [MyObj(4), MyObj(3), MyObj(2), MyObj(1), MyObj(0)]
sorted: [MyObj(0), MyObj(1), MyObj(2), MyObj(3), MyObj(4)]

屬性獲取方法類似于

lambda x, n='attrname':getattr(x,nz）

元素獲取方法類似于

lambda x,y=5:x[y]

from operator import *

l = [dict(val=-1*i) for i in xrange(4)]
print "dictionaries:", l
g = itemgetter("val")
vals = [g(i) for i in l]
print "values: ", vals
print "sorted:", sorted(l, key=g)

l = [(i,i*-2) for i in xrange(4)]
print "tuples: ", l
g = itemgetter(1)
vals = [g(i) for i in l]
print "values:", vals
print "sorted:", sorted(l, key=g)

結(jié)果如下：

dictionaries: [{'val': 0}, {'val': -1}, {'val': -2}, {'val': -3}]
values: [0, -1, -2, -3]
sorted: [{'val': -3}, {'val': -2}, {'val': -1}, {'val': 0}]
tuples: [(0, 0), (1, -2), (2, -4), (3, -6)]
values: [0, -2, -4, -6]
sorted: [(3, -6), (2, -4), (1, -2), (0, 0)]

除了序列之外，元素獲取方法還適用于映射。

結(jié)合操作符和定制類
operator模塊中的函數(shù)通過相應(yīng)操作的標(biāo)準(zhǔn)Python接口完成工作，所以它們不僅適用于內(nèi)置類型，還適用于用戶自定義類型。

from operator import *

class MyObj(object):
  def __init__(self, val):
    super(MyObj, self).__init__()
    self.val = val
    return 

  def __str__(self):
    return "MyObj(%s)" % self.val

  def __lt__(self, other):
    return self.val < other.val

  def __add__(self, other):
    return MyObj(self.val + other.val)

a = MyObj(1)
b = MyObj(2)

print lt(a, b)
print add(a,b)

結(jié)果如下所示：

True
MyObj(3)

類型檢查
operator 模塊還包含一些函數(shù)用來測(cè)試映射、數(shù)字和序列類型的API兼容性。

from operator import *

class NoType(object):
  pass

class MultiType(object):
  def __len__(self):
    return 0

  def __getitem__(self, name):
    return "mapping"

  def __int__(self):
    return 0

o = NoType()
t = MultiType()

for func in [isMappingType, isNumberType, isSequenceType]:
  print "%s(o):" % func.__name__, func(o)
  print "%s(t):" % func.__name__, func(t)

結(jié)果如下：

isMappingType(o): False
isMappingType(t): True
isNumberType(o): False
isNumberType(t): True
isSequenceType(o): False
isSequenceType(t): True

但是這些測(cè)試并不完善，因?yàn)榻杩跊]有嚴(yán)格定義。

獲取對(duì)象方法
使用methodcaller可以獲取對(duì)象的方法。

from operator import methodcaller

class Student(object):
  def __init__(self, name):
    self.name = name

  def getName(self):
    return self.name

stu = Student("Jim")
func = methodcaller('getName')
print func(stu)  # 輸出Jim

還可以給方法傳遞參數(shù)：

f=methodcaller('name', 'foo', bar=1)
f(b)  # return  b.name('foo', bar=1)
methodcaller方法等價(jià)于下面這個(gè)函數(shù)：

def methodcaller(name, *args, **kwargs):
   def caller(obj):
      return getattr(obj, name)(*args, **kwargs)
   return caller

您可能感興趣的文章: