Python MethodType方法详解和使用

引言

室友刷题，刷到一个python的优先队列的题目，py的优先队列无法自定义排序(无法理解), 然后尝试自己手写自定义的比较函数动态添加到堆节点的类上，于是查到了用method Type动态注入。特此记录一下。

python 中MethodType方法详解和使用
测试环境:python2.7

#!/usr/bin/python
# -*-coding:utf-8-*-

from types import MethodType 

# 首先看第一种方式

def set_age(self, arg):
    self.age = arg    

class Student(object):
    pass

#------以上为公共部分


s_one = Student()
#给student 创建一个方法 但这里不是在class中创建而是创建了一个链接把外部的set_age 方法用链接知道Student内
s_one.set_age = MethodType(set_age,s_one,Student)
s_one.set_age(32)  #调用实例方法
print s_one.age

#》》》》结果 32

s_two = Student() 
s_two.set_age(100)  #这里来验证下是在类内有方法还是类外有方法。
print s_two.age

#》》》》结果Traceback (most recent call last):
#》》》》 File “class2.py”, line 22, in
#》》》》 s_two.set_age(100) #这里来验证下是在类内有方法还是类外有方法。
#》》》》 AttributeError: ‘Student’ object has no attribute ‘set_age’

这里动态注入了实例，绑定了实例和方法，所以呢，另外的实例调用方法会报错，毕竟注入的是成员函数。

看另一种

#直接用类来创建一个方法  不过此时还是用链接的方式在类外的内存中创建
Student.set_age = MethodType(set_age,Student)
#此时在创建实例的时候外部方法 set_age 也会复制 这些实例和Student类都指向同一个set_age方法
new1 = Student()
new2 = Student()
new1.set_age(99)
new2.set_age(98) 　　#第二个会覆盖第一个 
print (new1.age,new2.age) 　　#看结果 2个都是98  

#》》》》(98, 98)

可以看到，这里没有注入实例，所有的调用返回相同的结果，相当于静态方法。

测试改成python3

报错原因： 动态注入的时候注入了类本身，而不是实例，所以当成了静态方法，而不存在静态的val。应该注入实例或。

AC代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None


from queue import PriorityQueue as PQ
import types


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

class Solution:
    def mergeKLists(self, lists) -> ListNode:
        if len(lists) == 0: return None
        re = ListNode(0)
        an = re
        pq = PQ()
        ListNode.__lt__ = lt
        for item in lists:
            if item:
                pq.put(item)
        while pq.qsize():
            a = pq.get()
            an.next = a
            an = a
            if a.next:
                pq.put(a.next)
        return re.next

总结: py3直接赋值就行，不需要methodTypes, 注意成员还是类级别的注入。