создать функцию для chi_square - PullRequest
Новая
       35

создать функцию для chi_square

0 голосов
/ 09 октября 2018

Мне нужно ввести функцию chi_square, и я застрял, потому что она всегда показывает, что при ее запуске неверный синтаксис. Интересно, как мне написать скрипт?И как мне ввести "v"? 

import numpy as np
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit

data = np.loadtxt("214 ohm.txt", skiprows=1)

xdata = [row[0] for row in data ]#x represents current unit is "V"
ydata = [row[1] for row in data]#y represents voltage unit is "mA"

percision_error_V =  np.array(xdata) * 0.0025 #we are using last digit of reading and multiply by measured voltage
accuracy_error_V = 0.01#we are using DC Vlotage, so use the error it provided online
erry = []

for i in range(len(percision_error_V)):
#to compare percision_error and accuracy_error for Voltage and use the larger one
erry.append(max(percision_error_V[i], accuracy_error_V))

def model_function (x, a, b):
    return a*x + b

p0 = [0 , 0.]#214ohm is measured by ohmeter

p_opt , p_cov = curve_fit ( model_function ,
                           xdata , ydata , p0,
                           erry , True )
print(erry)


a_opt = p_opt[0]
b_opt = p_opt[1]
print(p_cov)
print("diagonal of P-cov is",np.diag(p_cov))

print("a_opt, b_opt is ",a_opt, b_opt)
xhat = np.arange(0, 16, 0.1)

plt.plot(xhat, model_function(xhat, a_opt, b_opt), 'r-', label="model function")

plt.errorbar(xdata, ydata,np.array(erry),linestyle="",marker='s', label="error bar")
plt.legend()
plt.ylabel('Current (mA)')
plt.xlabel('Voltage(V)')
plt.title("Voltage vs. Current with 220ohm Resistor")
plt.show()

p_sigma = np.sqrt(np.diag(p_cov))
print("p_sigma is" ,p_sigma)
for i in range(len(xdata)):
    sum=sum((ydata[i]-model_function(xdata[i], a_opt, b_opt))
    chi.append(sum)

это обязательная функция, которую я должен поставить на python

Спасибо

мой код в порядке, пока уравнение хи-квадрат, интересно, как мне это исправить?

1 Ответ

0 голосов
/ 09 октября 2018

У вас есть отступ, 1 пропущенная скобка и имя переменной проблемы в этом примере кода

ОТ 

for i in range(len(xdata)):
sum=sum((ydata[i]-model_function(xdata[i], a_opt, b_opt))
1.append(sum)

TO 

for i in range(len(xdata)):
    sum=sum((ydata[i]-model_function(xdata[i], a_opt, b_opt)) )
    a.append(sum)

Переменные нельзя называть числами.Например, 1,2,3.Они должны начинаться со строки - a1, alfa, betta или s_t, _s.

...