Как сделать логарифмическую линию наилучшего соответствия? - PullRequest
0 голосов
/ 06 ноября 2018

Вот код, который у меня есть на данный момент, у меня есть некоторые данные, и я хочу, чтобы столбцы неопределенности остались. Какой кусок scipy мне нужен?

import matplotlib
import numpy as np
import matplotlib.pyplot as plt
import statistics
from statistics import stdev


y1 = [73.64, 76.47, 75.70, 71.71, 73.70, 79.39]
y2 = [219.70, 206.96, 235.31, 189.91, 256.48, 210.25]
y3 = [241.11, 271.70, 255.19, 229.67, 222.30, 200.70]
y4 = [256.59, 263.97, 262.17, 243.14, 245.42, 256.55]

y1_mean = statistics.mean(y1)
y2_mean = statistics.mean(y2)
y3_mean = statistics.mean(y3)
y4_mean = statistics.mean(y4)

y = np.array([y1_mean, y2_mean, y3_mean, y4_mean])
x = np.array([0,0.3,1.5,3])
e = np.array([stdev(y1), stdev(y2), stdev(y3), stdev(y4)])


plt.errorbar(x, y, e, linestyle = 'none', color = 'turquoise' )

plt.scatter(x, y, color = 'green')

plt.xlabel('x-coordinates')
plt.ylabel('y-coordinates')
plt.title('Sample graph')
plt.show()

Я хотел, чтобы это было так, но соответствовало моим данным:

The image

1 Ответ

0 голосов
/ 06 ноября 2018

Не совсем уверен, что вы хотите из-за плохого объяснения, но я постараюсь помочь вам, используя plt.semilogy() и кривую подгонку. Вы можете попробовать с plt.semilogy(x,y) и посмотреть, что вы получите, но в этом решении я хотел подогнать кривую, так что вот отредактированный код, надеюсь, он поможет вам или поможет решить вашу проблему:

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
import statistics as st
import scipy as sp
from scipy.optimize import curve_fit

y1 = [6,5,6,7,8]
y2 = [8,9,10,10,11]
y3 = [14,15,16,14,14]
y4 = [16,17,18,19,18]

y1_mean = statistics.mean(y1)
y2_mean = statistics.mean(y2)
y3_mean = statistics.mean(y3)
y4_mean = statistics.mean(y4)

y = np.array([y1_mean, y2_mean, y3_mean, y4_mean])
x = np.array([3,5,6,8])
e = np.array([st.stdev(y1), st.stdev(y2), st.stdev(y3), st.stdev(y4)])


def f(x,a,b,c): 
        return a*(np.square(x))+(b*x)+c

popt, pcov = curve_fit(f, x, y)

fitA = popt[0]
fitB = popt[1]
fitC = popt[0]

yFit = f(x,fitA,fitB,fitC)


plt.errorbar(x, y, e, linestyle = 'none', color = 'k', label="Points" )
plt.semilogy(x,y, color='g', linestyle="--", label="Log")
plt.semilogy(x,yFit,color='r', label="Log Fit")
plt.scatter(x, y, color = 'k')
plt.xlabel('x-coordinates')
plt.ylabel('y-coordinates')
plt.title('Sample graph')
plt.legend()
plt.show()

И вот что я получаю

enter image description here

...