Python:如何在给定数字数组的情况下绘制cdf函数

时间:2011-05-27 08:46:08

标签: python

我正在使用Windows。我只想输入一个数组并获取数组的cdf。

2 个答案:

答案 0 :(得分:4)

首先,您可以像这样实施CDF:

from bisect import bisect_left

class discrete_cdf:
    def __init__(self, data):
        self._data = data # must be sorted
        self._data_len = float(len(data))

    def __call__(self, point):
        return (len(self._data[:bisect_left(self._data, point)]) /
                self._data_len)

使用上面的类,您可以这样绘制:

from scipy.stats import norm
import matplotlib.pyplot as plt

cdf = discrete_cdf(your_data)
xvalues = range(0, max(your_data))
yvalues = [cdf(point) for point in xvalues]
plt.plot(xvalues, yvalues)

编辑arange没有任何意义,对于x和x + 1之间的所有点,cdf将始终相同。

答案 1 :(得分:2)

这就是你要追求的吗? 我提供了一个近似cdf的函数并绘制它。 (假设您要输入带有y值的pdf数组)

import matplotlib.pyplot as plt
from math import exp

xmin=0
xmax=5
steps=1000
stepsize=float(xmax-xmin)/float(steps)
xpoints=[i*stepsize for i in range(int(xmin/stepsize),int(xmax/stepsize))]
print xpoints,int(xmin/stepsize),int(xmax/stepsize)

ypoints=map(lambda x: exp(-x),xpoints)

def get_cdf(pdf_array):
    ans=[0]
    for i in range(0,len(pdf_array)-1):
        ans.append(ans[i]+(pdf_array[i]+pdf_array[i+1])/2.0*stepsize)
    return ans

cdfypoints=get_cdf(ypoints)

plt.plot(xpoints,ypoints)
plt.plot(xpoints,cdfypoints)
plt.show()

enter image description here