坐标的二维数组的逐元素交叉乘积



我正在使用一个数据集,该数据集将单位向量数组存储为向量组件的数组。

我将如何使用矢量化代码/广播来编写干净紧凑的代码,以逐个元素给出矢量的交叉乘积?

例如,这里有一个暴力方法,用于遍历数组的长度,挑选坐标,重新组合两个向量,然后计算叉积。

x = [0,0,1,1]
y = [0,1,0,1]
z = [1,0,0,1]
v1 = np.array([x,y,z])
x = [1,1,0,1]
y = [1,0,1,1]
z = [0,1,1,1]
v2 = np.array([x,y,z])
result = []
for i in range(0, len(x)):
a = [v1[0][i], v1[1][i], v1[2][i]]
b = [v2[0][i], v2[1][i], v2[2][i]]
result.append(np.cross(a,b))
result
>>>
[
array([-1,  1,  0]),
array([ 1,  0, -1]),
array([ 0, -1,  1]),
array([ 0,  0,  0])
]

我试图理解这个问题和答案来概括它,但失败了:
- 使用 Python 包含在 2 个数组中的向量的元素交叉乘积

np.cross也可以使用2D数组,您只需要指定正确的轴:

np.cross(v1,v2, axisa=0, axisb=0)
array([[-1,  1,  0],
[ 1,  0, -1],
[ 0, -1,  1],
[ 0,  0,  0]])

相关内容

  • 没有找到相关文章

最新更新