重复的Numpy阵列点产品

Question

我们假设我们有numpy数组A.shape = (1, 2)和B.shape = (2, 2)以及count X。

如果X = 1，我们需要结果= A.dot(B)

如果X = 2，我们需要结果= (A.dot(B)).dot(B)

如果X = 3，我们需要结果= ((A.dot(B)).dot(B)).dot(B)

我们如何在Python中编写一个函数，为我们提供基于9n X的正确结果？试图循环基于X的点积，但代码非常混乱，无法正常工作。

Answer 1

使用矩阵乘法交换和matrix_power的一个班轮：

A.dot(np.linalg.matrix_power(B, X))