python创建数组放入矩阵_python – 创建转换矩阵的numpy数组的正确方法是什么

这是一个直接而简单的方法：

c = np.cos(x_axis_rotations)

s = np.sin(x_axis_rotations)

matrices = np.zeros((len(x_axis_rotations), 3, 3))

matrices[:, 0, 0] = 1

matrices[:, 1, 1] = c

matrices[:, 1, 2] = -s

matrices[:, 2, 1] = s

matrices[:, 2, 2] = c

时间,对于好奇：

In [30]: angles = 2 * np.pi * np.random.rand(1000)

In [31]: timeit OP(angles)

100 loops, best of 3: 5.46 ms per loop

In [32]: timeit askewchan(angles)

10000 loops, best of 3: 39.6 µs per loop

In [33]: timeit divakar(angles)

10000 loops, best of 3: 93.8 µs per loop

In [34]: timeit divakar_oneline(angles)

10000 loops, best of 3: 56.1 µs per loop

In [35]: timeit divakar_combine(angles)

10000 loops, best of 3: 43.9 µs per loop

所有都比你的循环快得多,所以使用你最喜欢的:)