这段代码太难啦，@符号在Python中都有哪些作用？

Posted 2022-05-28 朝阳区靓仔_James

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了这段代码太难啦，@符号在Python中都有哪些作用？相关的知识，希望对你有一定的参考价值。

简单说@符号的作用有两个：装饰器和矩阵相乘。

装饰器

装饰器肯定是它常见的用法。这里给一个例子。这个看似简单的例子，其实知识点挺多的。如果你能轻松看明白，应该可以击败90%的Python学习者。

class Pizza(object):
    def __init__(self):
        self.toppings = []

    def __call__(self, topping):
        # 在方法定义前使用'@pizza实例'时，那个方法会被作为参数传给本方法
        self.toppings.append(topping())

    def __repr__(self):
        return str(self.toppings)

pizza = Pizza()

@pizza
def cheese():
    return 'cheese'
  
@pizza
def sauce():
    return 'sauce'

print(pizza)

输入结果：

['cheese', 'sauce']

矩阵相乘

在Python 3.5开始，Python引入了2个新的操作符和3个魔法函数，它们就是：

用于矩阵相乘的@和@=，后者类似于+=等符号，用于矩阵相乘并复制。
三个函数分别是：和@配套的__matmul__, rmatmul；和@=配套的__imatmul__

虽然Python支持了这两个操作符，并提供了相应的魔法函数，但Python内置库中并没有实现这些函数的类。

如果要用这些操作符，需要类库的开发者自己去实现，比如numpy就实现了@操作符，但它没有实现@=。

下面提供一个例子，我们来实现一下这个操作符：

class Matrix(list):
    def __matmul__(self, B):
        A = self
        return Matrix([[sum(A[i][k] * B[k][j] for k in range(len(B)))
                    for j in range(len(B[0])) ] for i in range(len(A))])
A = Matrix([[1, 2],[3, 4]])
B = Matrix([[5, 6],[7, 8]])
print(A @ B)