Numpy.hstack дает: TypeError: _vhstack_dispatcher() принимает 1 позиционный аргумент, но были заданы 2 ⇐ Python

[Anonymous]

У меня есть код Python, который принимает массив Nx2, в котором каждая строка представляет координаты [x,y] одной из N точек, и возвращает массив того же размера, содержащий преобразованные координаты [u,v] для каждой точки.

Код: Выделить всё

def inverse(arr):
return np.hstack([arr[:, 0:1], arr[:, 1:2] - arr[:, 0:1]**2])

Преобразование определяется как

Код: Выделить всё

u = x
v = y - x^2

Этот код работает нормально. Но потом у меня произошла трансформация

Код: Выделить всё

u = sqrt(x^2 + y^2)
v = arctan2(y, x)

И код

Код: Выделить всё

def inverse(arr):
return np.hstack(np.sqrt(arr[:, 0:1]**2 + arr[:, 1:2]**2), np.arctan2(arr[:, 1:2], arr[:, 0:1]))

Но этот код выдает следующую ошибку:

Код: Выделить всё

TypeError: _vhstack_dispatcher() takes 1 positional argument but 2 were given

В чем может быть причина этого? Как я могу это исправить?

Подробнее здесь: https://stackoverflow.com/questions/717 ... l-argument