Question

我必须将R^2数组X=[x1, x2]映射到R^3数组X'=[x1, x2, x1^2]中。

有没有紧凑的解决方案？

添加：我的起始数组是一个numpy数组，例如x = numpy.array([2, 3])。我想结束另一个numpy数组例如x' = numpy.array([2, 3, 4])。

Answer 1

对于您的数组是形状为(2,)的numpy数组，添加其他成员的一种方法是

xx = numpy.append(x, x[0]**2)

不幸的是，如果尝试使用x + [x[0]**2]之类的命令，numpy会将元素添加在一起，而不是附加新元素。如果您尝试使用类似[x[0], x[1], x[0]**2]的方法，则会得到一个列表而不是一个numpy数组。