arrays - 为什么 numpy 数组会以这种方式表现

Question

我正在做一个迷你项目，我打算使用纯 Python 而不需要任何外部库。但是在某个时刻，我得到了一个像这样的TypeErrorTypeError: unsupported operand type(s) for ** or pow(): 'list' and 'int'。这是因为试图直接将列表中的所有元素平方，如下所示：

my_list = [0, 7, 2, 9] 
print(my_list ** 2)

. 虽然我已经创建了一个可以将列表平方的函数，但是当我尝试像这样将一个 numpy 数组相乘时：

import numpy as np
my_list = np.array([0, 7, 2, 9])
print(my_list ** 2)# 打印数组[0, 49, 4, 9]

我的问题是为什么会发生这种情况？或者 numpy 是否有一个特殊的属性可以让它表现得像这样？

标签： arrayspython-3.xlistnumpy

Answer 1

np.ndarray为诸如__pow__. 这就是它如何设法按元素处理这些操作。使用内置列表，您可以使用列表推导：

[x**2 for x in my_list]

您还可以定义自己的类来公开此类自定义行为：

>>> class PowList(list):
...     def __pow__(self, other):
...         return [x**2 for x in self]
... 
>>> my_list = PowList([1, 2, 3, 4, 5])
>>> my_list ** 2
[1, 4, 9, 16, 25]