python - 根据numpy中的向量生成张量的元素

如何生成迭代向量不同值的numpy张量？例如，如果我想要以下公式：

rotation_mats=numpy.array([numpy.cos(theta),0,numpy.sin(theta)])

所有theta都替换为[0,pi/2]范围，结果是一个矩阵，如下所示：

[[1,0,0],
 [. . .],
 [0,0,1]]

最简单的方法是什么？

最佳答案

theta = numpy.arange(0, numpy.pi/2, 0.01) # You can control the step, of course
result = numpy.array([numpy.cos(theta),
                      numpy.zeros(len(theta)),
                      numpy.sin(theta)]
                     ).T

关于python - 根据numpy中的向量生成张量的元素，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/41450305/