У меня есть:
a = np.array([1, 2, 3])
Я хочу получить матрицу a_avg, где a_avg[i,j] - среднее значение a[i] и a[j], эффективно (без использования петель), такое, что результат:
a_avg
a_avg[i,j]
a[i]
a[j]
np.array([[1., 1.5, 2. ], [1.5, 2., 2.5], [2., 2.5, 3. ]])
Мое решение:
a_avg = (a[:, np.newaxis] + a) / 2
(я не нашел этот вопрос в Интернете, поэтому подумал, что его можно было бы поделиться)