[Python知识库]python魔法函数

声明：参考B站视频，自学成长记录
https://www.bilibili.com/video/BV1Cq4y1Q7Qv?p=7

什么是魔法函数？

python自定义的魔法函数
以双下划线开头 双下滑线结尾__XXX__格式

类对象是否可以被迭代？

如果直接for循环类实例对象会报 不可迭代
提示：TypeError: ‘For_Class’ object is not iterable

class For_Class(object):
    def __init__(self, employee_list):
        self.employee = employee_list

for_class = For_Class(['san', 'si', 'wu'])
for i in for_class :
    print(i)	# TypeError: 'For_Class' object is not iterable

因此引入我们第一个魔法函数__getitem__
让类或类对象可以被迭代

class Class_Getitem(object):
    def __init__(self, employee_list):
        self.employee = employee_list

    def __getitem__(self, item):
        return self.employee[item]

getitem = Class_Getitem(['san', 'si', 'wu'])
for i in getitem:
    print(i)

'''
运行结果：
	san
	si
	wu
'''

不同的魔法函数可以实现不同的特性增强

字符串表示 --> 魔法函数__str__

class Class_Str(object):
    def __init__(self, employee_list):
        self.employee = employee_list
        
	# 将对象格以字符串格式化输出
    def __str__(self):
        return ','.join(self.employee)


class_str = Class_Str(['san', 'si', 'wu'])
print(class_str)  # san,si,wu

字符串表示 --> 魔法函数__repr__

class Class_Repr(object):
    def __init__(self, employee_list):
        self.employee = employee_list
	
	# 开发者模式下
    def __repr__(self):
        return ','.join(self.employee)


class_repr = Class_Repr(['3', '4', '5'])
print(class_repr)

一元运算符 --> 魔法函数__abs__

class Class_Abs(object):
    def __init__(self, num):
        self.num = num

    # 获取绝对值
    def __abs__(self):
        return abs(self.num)

class_abs = Class_Abs(-5)
print(abs(class_abs))

算数运算符 --> 魔法函数__add__

class Class_Add(object):
    def __init__(self, x,y):
        self.x = x
        self.y = y
	
	# 相加
    def __add__(self, other):
        sums = Class_Add(self.x + other.x, self.y + other.y)
        return sums

    def __str__(self):
        return "x:{},y:{}".format(self.x,self.y)


class_add1 = Class_Add(1,2)
class_add2 = Class_Add(3,4)
print(class_add1 + class_add2)  # x:4,y:6

序列长度 --> 魔法函数__len__

class Class_Len(object):
    def __init__(self, employee_list):
        self.employee = employee_list
	
    def __len__(self):
        return len(self.employee)


class_len = Class_Len(['san', 'si', 'wu'])
print(len(class_len))  # 3