Как эффективно создать матрицу со средними значениями пар элементов вектора - PullRequest
0 голосов
/ 31 марта 2020

У меня есть:

a = np.array([1, 2, 3])

Я хочу получить матрицу 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. ]])

1 Ответ

0 голосов
/ 31 марта 2020

Мое решение:

a_avg = (a[:, np.newaxis] + a) / 2

(я не нашел этот вопрос в Интернете, поэтому подумал, что его можно было бы поделиться)

...