python - Numpy 2D array 行ごとの減算

Question

次の操作をベクトル化するのに苦労しています。x、y、z 距離の配列があり、各ベクトルの違いを見つける必要があります。

temp_result = np.array([[0.8, 0., 1.], [0., -0.6, 1.],[0.8, 0., 1.]])

私がやろうとしているのは、for ループの反復を使用せずに減算することです。

 temp_result[0] - temp_result[0]
 temp_result[0] - temp_result[1]
 temp_result[0] - temp_result[2]
 temp_result[1] - temp_result[0]
 temp_result[1] - temp_result[1]
 temp_result[1] - temp_result[2]
 temp_result[2] - temp_result[0]
 temp_result[2] - temp_result[1]
 temp_result[2] - temp_result[2]

ありがとう！

Answer 1

これが素晴らしいリシェイプベースのトリックです：

arr = temp_result
diffs = arr[:,None,:] - arr[None,:,:]

arr[i]次に、との間のベクトル差arr[j]がで見つかりdiffs[i,j]ます。

Answer 2

scipy.spatial.distance を確認してください。すべての距離とすべての距離の関数があります。