python - 使用 linalg.lstsq (numpy) 时出现数学域错误

Question

因此，我尝试在输入一组数据点时使用线性回归来显示趋势线。我正在使用 Tkinter 获取数据输入，然后将它们转换为 float 以将它们放入列表中。不过，我在运行程序时收到此错误代码。

    Exception in Tkinter callback
    Traceback (most recent call last):
      File "C:\Python27\lib\lib-tk\Tkinter.py", line 1486, in __call__
        return self.func(*args)
      File     "C:/Users/NIBO9901/PycharmProjects/Matteuppgift/Trendlinje/Input.py", line 78,  in plot
        m, c = np.linalg.lstsq(a, y)[0]
      File "C:\Python27\lib\site-packages\numpy\linalg\linalg.py", line 1889, in lstsq
        nlvl = max( 0, int( math.log( float(min(m, n))/2. ) ) + 1 )
    ValueError: math domain error

它引用的代码在这里:

    xList = []
    yList = []

    x = np.array(xList)
    y = np.array(yList)

    if 0 < len(inpX0.get()):
        xList.append(float(inpX0.get()))

    if 0 < len(inpX1.get()):
        xList.append(float(inpX1.get()))

    if 0 < len(inpY0.get()):
        yList.append(float(inpY0.get()))

    if 0 < len(inpY1.get()):
        yList.append(float(inpY1.get()))

    a = np.vstack([x, np.ones(len(x))]).T
    m, c = np.linalg.lstsq(a, y)[0]

    plt.plot(x, y, 'o', label='Data', markersize=10)
    plt.plot(x, m*x + c, 'r', label='Trendlinje')
    plt.legend()
    plt.show()

inpX/Y 是 Tkinter 条目。

Answer 1

numpy 数组是静态结构。您首先必须填充列表，然后将其转换为 numpy 数组。之后，对列表的修改将不再对 numpy 数组产生影响。

所以你想做的可能是:

xList = []
yList = []

if 0 < len(inpX0.get()):
    xList.append(float(inpX0.get()))

if 0 < len(inpX1.get()):
    xList.append(float(inpX1.get()))

if 0 < len(inpY0.get()):
    yList.append(float(inpY0.get()))

if 0 < len(inpY1.get()):
    yList.append(float(inpY1.get()))

a = np.vstack([x, np.ones(len(x))]).T
m, c = np.linalg.lstsq(a, y)[0]

# the lists are complete, now convert them to numpy arrays
x = np.array(xList)
y = np.array(yList)

plt.plot(x, y, 'o', label='Data', markersize=10)
plt.plot(x, m*x + c, 'r', label='Trendlinje')
plt.legend()
plt.show()