首页 > 解决方案 > 用 python 函数求解矢量投影

问题描述

创建一个名为计算 b 到向量 a 上的投影的函数。该函数应该适用于任何维度的向量。

条件:
- 向量 b 应该只包含一个并且应该与 a 具有相同的维度。

我的以下代码似乎不起作用。有什么改变的建议吗?

import numpy as np 

a = np.array([3,-1])

def computeProjection(a): 
    b = np.ones(np.size(a))
    projection = ((np.dot(a,b))*a) / (np.linalg.norm(a))**2
    assert np.size(a) == np.size(b)
    return projection()

computeProjection(a)

标签: python

解决方案


更改return projection()return projection
projection参数,不是函数。


推荐阅读