Python numpy.power() 用法与 LeetCode 求 n 次幂算法解析

清欢

发布时间：2026-10-14 | 2人浏览过

原创

本文详解Python中numpy.power()函数用于数组元素求n次方的用法，包括标量与数组混合运算。同时深入解析LeetCode中实现pow(x, n)的多种算法，涵盖暴力优化法、递归/迭代快速幂及位运算法，并介绍二分法求n次方的思路与代码实现。

示例如下：


numpy.power(x1, x2)

该函数用于计算数组中各元素的 n 次方。参数 x2 可以是单个数值，也可以是数组，但要求 x1 和 x2 的列数保持一致。


 >>> x1 = range(6)
 >>> x1
 [0, 1, 2, 3, 4, 5]
 >>> np.power(x1, 3)
 array([ 0,  1,  8, 27, 64, 125])


 >>> x2 = [1.0, 2.0, 3.0, 3.0, 2.0, 1.0]
 >>> np.power(x1, x2)
 array([ 0.,  1.,  8., 27., 16.,  5.])


 >>> x2 = np.array([[1, 2, 3, 3, 2, 1], [1, 2, 3, 3, 2, 1]])
 >>> x2
 array([[1, 2, 3, 3, 2, 1],
    [1, 2, 3, 3, 2, 1]])
 >>> np.power(x1, x2)
 array([[ 0, 1, 8, 27, 16, 5],
    [ 0, 1, 8, 27, 16, 5]])

补充阅读：Python 实现 pow 函数（求 n 次幂或 n 次方）。

类型一：求 n 次幂

目标是实现 pow(x, n)，即计算 x 的 n 次幂函数，其中 n 为整数。此问题对应 LeetCode 题目：pow函数的实现。

解法1：暴力法

这里的暴力法并非传统意义上的遍历，而是通过动态调整底数来加速计算过程。具体代码如下：


class Solution:
def myPow(self, x: float, n: int) -> float:
judge = True
if n<0:
n = -n
judge = False
if n==0:
return 1
final = 1 # 记录当前的乘积值
tmp = x # 记录当前的因子
count = 1 # 记录当前的因子是底数的多少倍
while n>0:
if n>=count:
final *= tmp
tmp = tmp*x
n -= count
count +=1
else:
tmp /= x
count -= 1
return final if judge else 1/final

解法2：根据奇偶幂分类（递归法、迭代法、位运算法）

当 n 为偶数时，pow(x, n) 等价于 pow(x^2, n/2)；

当 n 为奇数时，pow(x, n) 等价于 x * pow(x, n-1)。

递归方式的代码实现如下：


class Solution:
def myPow(self, x: float, n: int) -> float:
if n<0:
n = -n
return 1/self.help_(x,n)
return self.help_(x,n)
def help_(self,x,n):
if n==0:
return 1
if n%2 == 0: #如果是偶数
return self.help_(x*x, n//2)
# 如果是奇数
return self.help_(x*x,(n-1)//2)*x

迭代方式的代码如下：


class Solution:
def myPow(self, x: float, n: int) -> float:
judge = True
if n < 0:
n = -n
judge = False
final = 1
while n>0:
if n%2 == 0:
x *=x
n //= 2
final *= x
n -= 1
return final if judge else 1/final

Python 位运算符简介

该方法的逻辑与上述类似，区别在于利用位运算符判断奇偶性并执行除以 2 的操作（即移位操作）。代码如下：


class Solution:
def myPow(self, x: float, n: int) -> float:
judge = True
if n < 0:
n = -n
judge = False
final = 1
while n>0:
if n & 1: #代表是奇数
final *= x
x *= x
n >>= 1 # 右移一位
return final if judge else 1/final

类型二：求 n 次方

目标是实现 pow(x, n)，即计算 x 的 n 次方函数。约束条件为 x 大于 0，n 为大于 1 的整数。

解法：二分法求开方

核心思路是通过逐步逼近来寻找目标值。以 x 大于 1 的情况为例：

设定结果搜索范围为 [low, high]，初始设置 low=0, high = x，并假设结果 r=(low+high)/2；

若 r 的 n 次方大于 x，说明 r 取值偏大，保持 low 不变，更新 high= r，重新计算 r=(low+high)/2；

若 r 的 n 次方小于 x，说明 r 取值偏小，更新 low=r，保持 high 不变，重新计算 r=(low+high)/2；

具体代码如下：


class Solution:
def myPow(self, x: float, n: int) -> float:
# x为大于0的数,因为负数无法开平方(不考虑复数情况)
if x>1:
low,high = 0,x
else:
low,high =x,1
while True:
r = (low+high)/2
judge = 1
for i in range(n):
judge *= r
if x >1 and judge>x:break # 对于大于1的数，如果当前值已经大于它本身，则无需再算下去
if x <1 and judge
if abs(judge-x)<0.0000001: # 判断是否达到精度要求
print(pow(x,1/n)) # pow函数计算结果
return r
else:
if judge>x:
high = r
else:
low = r

以上为个人经验分享，希望能对读者有所帮助，也欢迎大家多多支持本站。若内容存在错误或未尽之处，敬请指正。