Question

我有以下一组指数;

ind = array([[array([0, 1, 4])],
           [array([0, 1, 2, 5])],
           [array([1, 2, 3, 6])],
           [array([2, 3, 7])],
           [array([0, 4, 5, 8])],
           [array([1, 4, 5, 6, 9])],
           [array([ 2,  5,  6,  7, 10])],
           [array([ 3,  6,  7, 11])],
           [array([ 4,  8,  9, 12])],
           [array([ 5,  8,  9, 10, 13])],
           [array([ 6,  9, 10, 11, 14])],
           [array([ 7, 10, 11, 15])],
           [array([ 8, 12, 13])],
           [array([ 9, 12, 13, 14])],
           [array([10, 13, 14, 15])],
           [array([11, 14, 15])]])

X的

，如下所示

X = array([[ 0.        ,  0.        ],
           [ 0.33333333,  0.        ],
           [ 0.66666667,  0.        ],
           [ 1.        ,  0.        ],
           [ 0.        ,  0.33333333],
           [ 0.33333333,  0.33333333],
           [ 0.66666667,  0.33333333],
           [ 1.        ,  0.33333333],
           [ 0.        ,  0.66666667],
           [ 0.33333333,  0.66666667],
           [ 0.66666667,  0.66666667],
           [ 1.        ,  0.66666667],
           [ 0.        ,  1.        ],
           [ 0.33333333,  1.        ],
           [ 0.66666667,  1.        ],
           [ 1.        ,  1.        ]])

有没有使用forloop来获得以下矩阵的有效方法;

A = [X[0,0]-X[0,0]  X[0,0]-X[1,0]  X[0,0]-X[4,0]
    X[1,0]-X[0,0]  X[1,0]-X[1,0]  X[1,0]-X[2,0] X[1,0]-X[5,0]
    X[2,0]-X[1,0]  X[2,0]-X[2,0]  X[2,0]-X[3,0] X[2,0]-X[6,0] 
    ...
    ...
    ...
   X[15,0]-X[11,0] X[15,0]-X[14,0] X[15,0]-X[15,0] ]

我已经尝试过np.subtract.outer和np.subtract。 ind结构非常难以使用（仅仅使用X[ind,0]）。我不想改变ind结构。在不使用forloop来实现矩阵A的情况下寻找切割方式。

感谢您的帮助

Answer 1

如果你转换索引数组，如：

ind = ind.flatten()

你可以获得你想做的事情：

a = X[:,0]
ans = [np.ones(len(i))*a[0]-a[i] for i in ind]

根据ind的原始定义，您必须改为使用i[0]。

编辑：这可以改进（如@Jaime所建议的那样）：

ans = [a[0] - a[i] for i in ind]

为数组的索引实现np.subtract.outer（[[array ....结构在python中

1 个答案: