Отображение массива в более высокий размер - Python

Question

Я должен отобразить R^2 массив X=[x1, x2] в R^3 массив X'=[x1, x2, x1^2].

Есть ли какое-нибудь компактное решение?

ДОБАВЛЕНО: мой начальный массив является массивом, например x = numpy.array([2, 3]). Я хочу закончить с другим массивом Numpy, например. x' = numpy.array([2, 3, 4]).

Answer 1

Для случая, когда ваш массив представляет собой пустой массив с формой (2,), один из способов добавить дополнительный член -

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

К сожалению, если вы попробуете что-то вроде x + [x[0]**2], numpy добавит элементы вместе, а не добавит новый элемент. Если вы попробуете что-то вроде [x[0], x[1], x[0]**2], вы получите список, а не пустой массив.