問題

我必須將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]).

  最佳答案

對於陣列是一個帶有形狀(2,)的numpy陣列的情況,新增額外成員的一種方法是

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

不幸的是,如果嘗試像x + [x[0]**2] numpy這樣的東西,numpy將新增元素而不是附加新元素.如果嘗試像[x[0], x[1], x[0]**2]這樣的東西,你最終得到一個列表而不是一個numpy陣列.

  相同標籤的其他問題

pythonarraystransformdimensions