[Numpy-discussion] simple vector->matrix question

Warren Weckesser warren.weckesser at enthought.com
Thu Oct 6 08:18:20 EDT 2011


On Thu, Oct 6, 2011 at 7:08 AM, Neal Becker <ndbecker2 at gmail.com> wrote:

> Given a vector y, I want a matrix H whose rows are
>
> y - x0
> y - x1
> y - x2
> ...
>
>
> where x_i are scalars
>
> Suggestion?
>
>

In [15]: import numpy as np

In [16]: y = np.array([10.0, 20.0, 30.0])

In [17]: x = np.array([0, 1, 2, 4])

In [18]: H = y - x[:, np.newaxis]

In [19]: H
Out[19]:
array([[ 10.,  20.,  30.],
       [  9.,  19.,  29.],
       [  8.,  18.,  28.],
       [  6.,  16.,  26.]])


Warren



> _______________________________________________
> NumPy-Discussion mailing list
> NumPy-Discussion at scipy.org
> http://mail.scipy.org/mailman/listinfo/numpy-discussion
>
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/numpy-discussion/attachments/20111006/c3c7e427/attachment.html>


More information about the NumPy-Discussion mailing list