我想得到形状为[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
如何修改我的代码？
我找到了一个解决方案。不应使用列表理解：
X=np.vstack([df.X,np.ones(len(df.X))]).T
Y = df.Y
m,b = np.linalg.lstsq(X,Y)[0]
这对我有用。