python - 转换一个numpy数组

Question

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

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

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

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

Answer 1

像这样:

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 功能的小技巧。它们有点难看，但可以让你写得更紧凑一些。