Точечный продукт каждого элемента с каждым другим элементом массива - PullRequest
1 голос
/ 17 января 2011

Есть ли простой способ объединить точечное произведение одного элемента массива с другим?Итак, учитывая:

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

Я хотел бы получить результат:

array([  32.,   50.,  122.])

Т.е. [0] точка a [1], a [0] точка a [2],a [1] точка a [2].

Массив, с которым я работаю, НЕ будет квадратным;это всего лишь пример.

Спасибо!

Ответы [ 3 ]

4 голосов
/ 17 января 2011
>>> X = scipy.matrix('1 2 3; 4 5 6; 7 8 9')
>>> X*X.T
matrix([[ 14,  32,  50],
        [ 32,  77, 122],
        [ 50, 122, 194]])

Это дает вам больше, чем вы хотели, но это, несомненно, легко.

Или

>>> X = scipy.array([[1,2,3], [4,5,6], [7,8,9]])
>>> scipy.dot(X, X.T)
array([[ 14,  32,  50],
       [ 32,  77, 122],
       [ 50, 122, 194]])
1 голос
/ 17 января 2011

Так как похоже, что вы используете numpy:

from itertools import combinations
import numpy as np

dot_products = [np.dot(*v) for v in combinations(vectors, 2)]

Я проверил это, и похоже, что оно работает на моей установке Python.

0 голосов
/ 17 января 2011

Вот еще один:

>>> a = numpy.array([[1, 2, 3],
...        [4, 5, 6],
...        [7, 8, 9]])
>>> numpy.array([numpy.dot(a[i], a[j]) for i in range(len(a)) for j in range(i + 1, len(a))])
array([ 32,  50, 122])
...