将数据帧中的列拟合为动画直方图

2024-05-13 07:09:39 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图生成一个动画直方图,它使用我创建的数据帧中的数据行。下面是我用来生成直方图的代码。代码使用 data = np.random.randn(1000)工作,但当我将直方图替换为data = df['GDP']时,它不会设置直方图的动画,而是输出一个未设置动画的直方图。我正在尝试将数据帧中的一列放入此代码中:

import matplotlib.pyplot as plt
import matplotlib.patches as patches
import matplotlib.path as path
import matplotlib.animation as animation

fig, ax = plt.subplots()

# histogram our data with numpy
data = np.random.randn(1000)
n, bins = np.histogram(data, 100)

# get the corners of the rectangles for the histogram
left = np.array(bins[:-1])
right = np.array(bins[1:])
bottom = np.zeros(len(left))
top = bottom + n
nrects = len(left)

# here comes the tricky part -- we have to set up the vertex and path
# codes arrays using moveto, lineto and closepoly

# for each rect: 1 for the MOVETO, 3 for the LINETO, 1 for the
# CLOSEPOLY; the vert for the closepoly is ignored but we still need
# it to keep the codes aligned with the vertices
nverts = nrects*(1+3+1)
verts = np.zeros((nverts, 2))
codes = np.ones(nverts, int) * path.Path.LINETO
codes[0::5] = path.Path.MOVETO
codes[4::5] = path.Path.CLOSEPOLY
verts[0::5,0] = left
verts[0::5,1] = bottom
verts[1::5,0] = left
verts[1::5,1] = top
verts[2::5,0] = right
verts[2::5,1] = top
verts[3::5,0] = right
verts[3::5,1] = bottom

barpath = path.Path(verts, codes)
patch = patches.PathPatch(barpath, facecolor='green', edgecolor='yellow', alpha=0.5)
ax.add_patch(patch)

ax.set_xlim(left[0], right[-1])
ax.set_ylim(bottom.min(), top.max())

def animate(i):
    # simulate new data coming in
    data = np.random.randn(1000)
    n, bins = np.histogram(data, 100)
    top = bottom + n
    verts[1::5,1] = top
    verts[2::5,1] = top

ani = animation.FuncAnimation(fig, animate, 100, repeat=False)



from IPython.display import HTML
HTML(ani.to_jshtml())

为了适应我自己的数据,我正在替换:

# histogram our data with numpy
data = np.random.randn(1000)

以及:

 # simulate new data coming in
 data = np.random.randn(1000)

在我的数据框中有一列包含247行:

data = df['GDP']

输出是一个带有我自己数据的直方图,但是它不像data = np.random.randn(1000)那样具有动画效果


Tags: the数据pathimportfordatatopnp