錯誤適合

Question

代碼：有一個與它似乎是5日線，特別是與np.polyfit問題，

i = text_i.get() 
r = text_r.get() 
i = i.split(", ") 
r = r.split(", ")   

fit = np.polyfit(i,r,1) 
fit_fn = np.poly1d(fit) 
plt.plot(i,r, 'yo', i, fit_fn(i), '--k') 
plt.ilim(0, 5) 
plt.rlim(0, 12) 

錯誤消息

fit = np.polyfit(i,r,1) 
    File "C:\Python27\lib\site-packages\numpy\lib\polynomial.py", line 546, in polyfit 
    x = NX.asarray(x) + 0.0 
TypeError: ufunc 'add' did not contain a loop with signature matching types 
dtype('S32') dtype('S32') dtype('S32') 

請找什麼代碼錯了？

Answer 1

np.polyfit(x,y,1)需要一個數值數據列表或數組作爲其參數x和y的輸入。然而，你輸入的是一些字符串。因此，您需要拆分該字符串，並將其每個元素轉換爲數字，然後將其傳遞到polyfit。 嘗試：

i = np.array(list(map(float, i.split(", "))))