python - np.linalg.lstsq(X，Y)[0]-TypeError : No loop matching the specified signature and casting was found for ufunc lstsq_n

Question

我想以[X 1]的形式获取X值。
为此，我使用以下代码:

X = np.array([[value,1] for value in X])

我得到这个警告...

/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:2: VisibleDeprecationWarning: Creating an ndarray from ragged nested sequences (which is a list-or-tuple of lists-or-tuples-or ndarrays with different lengths or shapes) is deprecated. If you meant to do this, you must specify 'dtype=object' when creating the ndarray

...但是似乎可行。
但是当我随后尝试使用thgis代码获取m和b值时:

m, b = np.linalg.lstsq(X,Y)[0]

我收到此错误:

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-32-761e189a9409> in <module>()
----> 1 m, b = np.linalg.lstsq(X,Y)[0]

<__array_function__ internals> in lstsq(*args, **kwargs)

/usr/local/lib/python3.6/dist-packages/numpy/linalg/linalg.py in lstsq(a, b, rcond)
   2304         # lapack can't handle n_rhs = 0 - so allocate the array one larger in that axis
   2305         b = zeros(b.shape[:-2] + (m, n_rhs + 1), dtype=b.dtype)
-> 2306     x, resids, rank, s = gufunc(a, b, rcond, signature=signature, extobj=extobj)
   2307     if m == 0:
   2308         x[...] = 0

TypeError: No loop matching the specified signature and casting was found for ufunc lstsq_n

如何修改我的代码？

Answer 1

我找到了解决方案。不应使用列表理解:

X=np.vstack([df.X,np.ones(len(df.X))]).T

Y = df.Y

m,b = np.linalg.lstsq(X,Y)[0]

这对我有用。