在没有循环的numpy中计算xi-xj矩阵(通过api调用)

Question

如何在没有循环的numpy中计算xi-xj矩阵(通过api调用)？

这是从什么开始:

import numpy as np
x = np.random.rand(4)
xij = np.matrix([xi-xj for xj in x for xi in x]).reshape(4,4)

Answer 1

您可以利用广播x从x平面数组中减去列向量并生成矩阵.

>>> x = np.random.rand(4)

然后:

>>> x - x[:,np.newaxis]
array([[ 0.        ,  0.89175647,  0.80930233,  0.37955823],
       [-0.89175647,  0.        , -0.08245415, -0.51219825],
       [-0.80930233,  0.08245415,  0.        , -0.4297441 ],
       [-0.37955823,  0.51219825,  0.4297441 ,  0.        ]])

如果你想要一个矩阵对象(而不是默认的数组对象),你可以写:

np.matrix(x - x[:,np.newaxis])