Question

这是我的代码。我想要它返回的是一个矩阵数组

[[1,1],[1,1]], [[2,4],[8,16]], [[3,9],[27,81]]

我知道我可以使用for循环和循环遍历我的向量k，但我想知道是否有一种我想念的简单方法。谢谢！

from numpy import *
import numpy as np

k=np.arange(1,4,1)
print k

def exam(p):
   return np.array([[p,p**2],[p**3,p**4]])

print exam(k)

输出：

[1 2 3]

[[[ 1  2  3]

  [ 1  4  9]]

 [[ 1  8 27]

  [ 1 16 81]]]

Answer 1

关键是要玩形状和广播。

b = np.arange(1,4) # the base
e = np.arange(1,5) # the exponent

b[:,np.newaxis] ** e
=> 
array([[ 1,  1,  1,  1],
       [ 2,  4,  8, 16],
       [ 3,  9, 27, 81]])

(b[:,None] ** e).reshape(-1,2,2) 
=>
array([[[ 1,  1],
        [ 1,  1]],

       [[ 2,  4],
        [ 8, 16]],

       [[ 3,  9],
        [27, 81]]])

如果必须将输出作为矩阵列表，请执行：

m = (b[:,None] ** e).reshape(-1,2,2)
[ np.mat(a) for a in m ]
=>
[matrix([[1, 1],
        [1, 1]]),
 matrix([[ 2,  4],
        [ 8, 16]]),
 matrix([[ 3,  9],
        [27, 81]])]

一组矩阵，使用Python中的数组值构建

1 个答案: