python绘制剖面图_用python绘制剖面图

我自己为这个功能做了一个模块。在import pandas as pd

from pandas import Series, DataFrame

import numpy as np

import matplotlib.pyplot as plt

def Profile(x,y,nbins,xmin,xmax,ax):

df = DataFrame({'x' : x , 'y' : y})

binedges = xmin + ((xmax-xmin)/nbins) * np.arange(nbins+1)

df['bin'] = np.digitize(df['x'],binedges)

bincenters = xmin + ((xmax-xmin)/nbins)*np.arange(nbins) + ((xmax-xmin)/(2*nbins))

ProfileFrame = DataFrame({'bincenters' : bincenters, 'N' : df['bin'].value_counts(sort=False)},index=range(1,nbins+1))

bins = ProfileFrame.index.values

for bin in bins:

ProfileFrame.ix[bin,'ymean'] = df.ix[df['bin']==bin,'y'].mean()

ProfileFrame.ix[bin,'yStandDev'] = df.ix[df['bin']==bin,'y