4

我有一个这种形式的numpy数组

[[-0.77947021  0.83822138]
[ 0.15563491  0.89537743]
[-0.0599077  -0.71777995]
[ 0.20759636  0.75893338]]

我想从原始数组列表[x1, x2, x1*x2]中创建这种形式的 numpy 数组。[x1, x2]

目前我正在使用 python 代码创建一个列表列表，然后将其转换为 numpy 数组。但我认为可能有更好的方法来做到这一点。

4

2 回答 2

5

像这样：

In [22]: import numpy as np

In [23]: x = np.array([[-0.77947021,  0.83822138],
    ...: [ 0.15563491,  0.89537743],
    ...: [-0.0599077,  -0.71777995],
    ...: [ 0.20759636,  0.75893338]])

In [24]: np.c_[x, x[:,0] * x[:,1]]
Out[24]: 
array([[-0.77947021,  0.83822138, -0.6533686 ],
       [ 0.15563491,  0.89537743,  0.13935199],
       [-0.0599077 , -0.71777995,  0.04300055],
       [ 0.20759636,  0.75893338,  0.15755181]])

这使用numpy.c_, 这是一个方便的函数，可以沿它们的第二维连接各种数组。

您可以在Numpy 的教程中找到有关连接数组的更多信息。hstack（参见 Jaime 的回答）、、、vstack和concatenate等row_stack函数可能column_stack是您应该使用的“官方”函数。函数r_和c_有点破解来模拟 Matlab 的一些功能。它们有点难看，但可以让你写得更紧凑。

于 2013-11-07T17:51:15.943 回答
1

有一百万种不同的方式，我可能会这样做：

>>> x = np.random.rand(5, 2)
>>> np.hstack((x, np.prod(x, axis=1, keepdims=True)))
array([[ 0.39614232,  0.14416164,  0.05710853],
       [ 0.75068436,  0.61687739,  0.46308021],
       [ 0.90181541,  0.20365294,  0.18365736],
       [ 0.08172452,  0.36334486,  0.02969418],
       [ 0.61455203,  0.80694432,  0.49590927]])
于 2013-11-07T17:58:42.777 回答