gpt4 book ai didi

python - linalg.norm 不采用轴参数

转载 作者:太空狗 更新时间:2023-10-29 21:24:28 25 4
gpt4 key购买 nike

我在 Pyzo 中使用 Python 3。请告诉我为什么 linalg.norm 函数不能识别 axis 参数。

这段代码:

c = np.array([[ 1, 2, 3],[-1, 1, 4]])
d=linalg.norm(c, axis=1)

返回错误:

TypeError: norm() got an unexpected keyword argument 'axis'

最佳答案

linalg.norm 不接受 axis 参数。你可以通过以下方式解决这个问题:

np.apply_along_axis(np.linalg.norm, 1, c)
# array([ 3.74165739, 4.24264069])

或者为了更快,自己实现:

np.sqrt(np.einsum('ij,ij->i',c,c))
# array([ 3.74165739, 4.24264069])

对于时间:

timeit np.apply_along_axis(np.linalg.norm, 1, c)
10000 loops, best of 3: 170 µs per loop

timeit np.sqrt(np.einsum('ij,ij->i',c,c))
100000 loops, best of 3: 10.7 µs per loop

关于python - linalg.norm 不采用轴参数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/20168596/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com