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

[Anonymous]

У меня есть этот код Python, который принимает в массиве Nx2 Numpy, в котором каждая строка представляет собой координаты [x, y] одной из N -точек, и она возвращает массив одного и того же размера, содержащего преобразованные [u, v] координаты для каждой точки.def inverse(arr):
return np.hstack([arr[:, 0:1], arr[:, 1:2] - arr[:, 0:1]**2])
< /code>
Преобразование определяется как < /p>
u = x
v = y - x^2
< /code>
Этот код работает нормально. Но тогда у меня есть преобразование < /p>
u = sqrt(x^2 + y^2)
v = arctan2(y, x)
< /code>
и код < /p>
def inverse(arr):
return np.hstack(np.sqrt(arr[:, 0:1]**2 + arr[:, 1:2]**2), np.arctan2(arr[:, 1:2], arr[:, 0:1]))
< /code>
Но этот код дает следующую ошибку: < /p>
TypeError: _vhstack_dispatcher() takes 1 positional argument but 2 were given
< /code>
Что может быть причиной этого? Как это исправить?

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