问Python:生成“曲线拟合分数”

EN

from PIL import Image

def populateData(picture) :
    """Open an image and populate a list of lists with the grayscale value"""
    im = Image.open(picture)           
    size = im.size                    
    width = size[0]                   
    height = size[1]                   
    allPixels = list(im.getdata())      
    pixelList = [allPixels[width*i :    
                       width * (i+1)] for i in range(height)] 
    return(pixelList)

rawData = populateData("testTop.tif")   

import numpy as np
from scipy.optimize import curve_fit

def findVertex(listOfRows) :

    xList = []
    for row in listOfRows :
        x = np.arange(len(row))
        ffunc = lambda x, a, x0, s: a*np.exp(-0.5*(x-x0)**2/s**2) 
        p, _ = curve_fit(ffunc, x, row, p0=[100,5,2])
        x0 = p[1] 
        xList.append(x0)
    xArray = np.array(xList)    
    return(xArray)

xValues = findVertex(rawData)

def buildRows(listOfRows) :
    yArray = np.arange(len(listOfRows))
    return(yArray)

yValues = buildRows(rawData)

from matplotlib import pyplot as plt
from scipy import ndimage

image = ndimage.imread("testTop.tif",flatten=True)
fig = plt.figure()
axes = fig.add_subplot(111)
axes.imshow(image)
axes.plot(xValues, yValues, 'k-')

axes.set_title('testLine')
axes.grid()
axes.set_xlabel('x')
axes.set_ylabel('time')
plt.show()