Início / coding / Perguntas / 77838692
Accepted
G-09
Asked: 2024-01-18 18:34:32 +0800 CST

Numpy calcula o produto das diferenças de duas matrizes em relação a todas as linhas da matriz

  • 772

o seguinte é o que eu quero alcançar. Existe uma maneira de fazer isso de forma eficiente com Numpy ou outros métodos em Python? Já que meus dados seriam grandes. Eu gostaria de fazer isso de forma mais eficiente. Agradeço antecipadamente.

import numpy as np
x = np.arange(6).reshape(3,2)
y = np.arange(10).reshape(5,2)
z = np.zeros((len(x), len(y)))

for i in range(len(z)):
    for j in range(len(z[0])):
        z[i,j] = np.prod(x[i] - y[j])
numpy

1 respostas

  1. Best Answer

    Basta vetorizar seu código:

    z = (x[:,None]-y).prod(axis=-1)

    Saída:

    array([[ 0,  4, 16, 36, 64],
       [ 4,  0,  4, 16, 36],
       [16,  4,  0,  4, 16]])

    Intermediário:

    (x[:,None]-y).

array([[[ 0,  0],
        [-2, -2],
        [-4, -4],
        [-6, -6],
        [-8, -8]],

       [[ 2,  2],
        [ 0,  0],
        [-2, -2],
        [-4, -4],
        [-6, -6]],

       [[ 4,  4],
        [ 2,  2],
        [ 0,  0],
        [-2, -2],
        [-4, -4]]])
    • 1

relate perguntas