我有一个df,它包含一个鼠标鼻子的x,y坐标,我想用它来制作动画散点图。目前,我有静态散点图的代码。
import pandas as pd import numpy as np import matplotlib.pyplot as plt import matplotlib.animation as animation import os from pathlib import Path from IPython.display import HTML #import video pose estimation data video='topNoF.mp4' DLCscorer='C:/Users/Bri-Guy/Desktop/DLC/h5/topFDLC_resnet50_topAnalysisJan20shuffle1_100000' dataname = DLCscorer+'.h5' #can change to .csv instead; make sure to change pd.read_hdf() to pd.read_csv() df=pd.read_hdf(os.path.join(dataname)) #get X & Y coordinates of snout bdp scorer=df.columns.get_level_values(0)[0] x=df[scorer]['snout']['x'].values y=df[scorer]['snout']['y'].values #produce a static scatterplot trace of snout movement length=len(x) n = len(x) color = [] for i in range (1,n+1): color.append(i/n) scatter=plt.scatter(x,y,c=color, cmap='inferno') ax = scatter.axes
从而产生
我想使用此Stack Exchange答案中的代码变体:Matplotlib Plot Points Over Time Where Old Points Fade以如下方式设置散点图的动画:
import numpy as np import matplotlib.pyplot as plt from matplotlib.animation import FuncAnimation from IPython.display import HTML from matplotlib.colors import LinearSegmentedColormap plt.rcParams['animation.ffmpeg_path'] = r'C:/Users/Bri-Guy/anaconda3/envs/pWGA/Library/bin/ffmpeg.exe' x=df[scorer]['snout']['x'].values y=df[scorer]['snout']['y'].values fig, ax = plt.subplots() plt.xlim(0, max(x)+100) plt.ylim(0, max(y)+100) graph, = plt.plot([], [], 'o') def animate(i): graph.set_data(x[:i+1], y[:i+1]) return graph ani = FuncAnimation(fig, animate, frames=len(y), interval=20) html = ani.to_html5_video() HTML(html)
理想情况下,我希望此动画中的点随着时间推移而褪色(像链接的示例一样平滑褪色)。此外,我不确定如何为动画散点图设置颜色贴图(如inferno)。淡化(&A);设置颜色图的能力是我最重要的优先事项。
我使用链接示例中的代码遇到的主要问题是以下方面:
def get_new_vals(): n = np.random.randint(1,5) x = np.random.rand(n) y = np.random.rand(n) return list(x), list(y) def update(t): global x_vals, y_vals, intensity # Get intermediate points new_xvals, new_yvals = get_new_vals() x_vals.extend(new_xvals) y_vals.extend(new_yvals) # Put new values in your plot scatter.set_offsets(np.c_[x_vals,y_vals]) #calculate new color values intensity = np.concatenate((np.array(intensity)*0.96, np.ones(len(new_xvals)))) scatter.set_array(intensity) # Set title ax.set_title('Time: %0.3f' %t)
我需要从x=df[scorer]['snout']['x'].values
&y=df[scorer]['snout']['y'].values
. 必须按x[0]
到x[len(x)-1]
的顺序调用这些值,以便绘图按时间顺序更新。然而,当试图在get_new_vals(i)
中添加参数时,我会出错,因为我认为动画的t
变量不是整数。我不确定数组x&;因为它们是浮点。
我提前感谢你的帮助!如果我能为你澄清什么,请告诉我。下面我发布了x,y变量中的一些数据:
print(x[:200]) [276.89370728 280.57974243 285.25439453 285.55096436 284.71258545 283.52386475 284.31976318 285.08609009 285.56118774 285.38183594 289.21246338 295.28497314 303.41043091 315.51828003 324.87826538 333.36367798 338.73730469 341.11685181 349.20300293 357.63671875 366.72702026 395.68356323 385.37298584 387.86871338 387.58526611 382.20205688 378.13674927 373.97241211 368.39953613 591.94116211 347.27310181 616.52069092 608.12902832 605.11340332 602.10974121 598.72052002 598.48504639 599.19256592 601.30432129 603.32104492 604.9621582 605.21533203 621.36779785 627.51617432 626.20269775 621.00164795 618.92498779 617.44885254 615.73883057 598.8916626 594.17883301 593.38647461 589.3248291 592.67895508 593.67053223 589.05767822 589.08850098 303.0085144 568.39239502 555.08520508 550.79425049 547.77197266 547.21954346 313.01544189 333.96121216 348.59899902 353.26141357 358.76705933 360.81588745 363.94262695 527.38165283 522.80316162 518.20489502 521.84442139 525.30664062 526.43286133 532.38995361 536.35961914 536.51574707 540.41906738 545.77844238 545.22381592 545.34112549 540.22357178 537.93457031 534.03442383 532.6651001 522.52618408 505.38290405 489.37664795 469.75460815 448.28039551 424.54315186 403.87719727 383.25265503 359.93307495 335.06869507 318.53125 296.43450928 288.86499023 442.87780762 443.70950317 440.31652832 439.50854492 439.11328125 434.4161377 216.55622864 216.7456665 215.32554626 213.63644409 213.8143158 213.96568298 213.87882996 214.10801697 214.05957031 212.54373169 216.25740051 214.80444336 216.47532654 218.31072998 215.78303528 213.40249634 292.92352295 290.80630493 287.03222656 283.45129395 390.10848999 274.83648682 269.53741455 215.76341248 218.75086975 220.38156128 219.87997437 219.83804321 218.52023315 216.93737793 218.18110657 218.31959534 224.79884338 224.69064331 221.88998413 218.67016602 216.9510498 216.63031006 215.88612366 217.3243103 217.01783752 214.08659363 213.87808228 211.14770508 206.47595215 214.88208008 214.39358521 212.50665283 212.39123535 213.95169067 217.72639465 313.20504761 288.23443604 283.30273438 283.1756897 281.46990967 276.20397949 273.39535522 274.38088989 267.42678833 269.19915771 271.11810303 331.80795288 330.61746216 329.03930664 227.14578247 226.81338501 227.80999756 229.65690613 229.97644043 207.91325378 219.31289673 225.3374939 230.50515747 313.22525024 310.83474731 306.02667236 299.73217773 288.91854858 278.45489502 266.55349731 264.91229248 256.15029907 249.70783997 245.47244263 246.11851501 245.35572815 246.03157043 246.50708008 246.63691711 245.77215576 245.09873962 241.44792175 243.87677002] print(y[:200]) [1321.18652344 1316.84301758 1316.04064941 1315.66455078 1315.38586426 1315.74560547 1317.04711914 1318.11218262 1320.09631348 1321.45703125 1328.39794922 1339.04956055 1353.28076172 1364.76757812 1371.4901123 1376.57568359 1381.58361816 1383.17236328 1390.66870117 1401.93652344 1412.72961426 1393.1583252 1437.81616211 1440.15380859 1440.01843262 1438.75891113 1437.50012207 1436.55932617 1429.39416504 1620.10900879 1400.08654785 1506.79125977 1497.49243164 1491.12316895 1488.84606934 1487.79296875 1489.73840332 1488.69665527 1490.09631348 1490.75927734 1490.99291992 1487.58154297 1610.09851074 1612.06115723 1613.1282959 1615.44006348 1619.22009277 1620.50915527 1618.51501465 1495.30175781 1496.44470215 1497.99816895 1492.10546875 1474.48339844 1481.59130859 1475.67248535 1477.48083496 1375.00964355 1480.91625977 1486.29675293 1493.73193359 1501.6204834 1507.31176758 1366.4050293 1358.88977051 1346.67797852 1339.04455566 1333.7755127 1328.54821777 1322.38439941 1502.17102051 1497.28198242 1493.86022949 1479.5690918 1469.38928223 1458.9095459 1448.56921387 1448.53613281 1443.38757324 1444.77124023 1443.64233398 1440.03857422 1431.31848145 1427.53820801 1427.53393555 1428.34887695 1429.39526367 1431.30249023 1440.09594727 1457.33044434 1474.85864258 1492.55554199 1498.40905762 1497.65649414 1503.4185791 1507.8626709 1507.43164062 1506.86315918 1507.16052246 1506.81115723 1313.21789551 1317.2467041 1322.02380371 1320.96960449 1314.60107422 1314.20336914 1640.76843262 1656.12866211 1667.98669434 1675.00683594 1675.64379883 1678.5567627 1678.24157715 1673.58984375 1667.44628906 1651.3112793 1635.78564453 1626.48413086 1616.28967285 1594.74316406 1580.41320801 1579.52124023 1506.03894043 1504.37646484 1501.6105957 1500.49902344 1316.76806641 1491.15759277 1471.89294434 1575.55322266 1585.22619629 1594.93151855 1598.60961914 1597.25134277 1598.54443359 1599.59936523 1597.15466309 1594.58544922 1585.51281738 1582.63903809 1584.89025879 1585.41784668 1589.9654541 1592.00085449 1596.66369629 1598.4128418 1598.8269043 1600.46069336 1596.1394043 1597.69421387 1598.36975098 1582.69897461 1580.45617676 1581.58618164 1580.9498291 1578.98144531 1575.28918457 1494.18151855 1490.47180176 1490.87109375 1485.41467285 1480.73291016 1481.30554199 1485.26867676 1489.50756836 1497.07263184 1501.99645996 1505.94543457 1267.27172852 1270.28515625 1281.31738281 1568.67248535 1576.60119629 1578.63500977 1577.04748535 1577.89709473 1681.06103516 1681.95227051 1684.63745117 1686.96484375 1325.62084961 1331.21569824 1331.70007324 1331.41809082 1336.49401855 1345.08703613 1351.00073242 1356.37145996 1360.50805664 1367.5177002 1371.69470215 1376.6151123 1377.27844238 1376.92382812 1376.37487793 1377.03894043 1377.87390137 1378.53796387 1373.02990723 1365.00915527]