其他分享
首页 > 其他分享> > numpy fft 备忘.

numpy fft 备忘.

作者:互联网

主要参考以下注释网站

  1 #encoding:utf-8
  2 #https://blog.csdn.net/weixin_39949473/article/details/111337230
  3 #https://blog.csdn.net/weixin_35756373/article/details/112713361
  4 from numpy.fft import fft 
  5 
  6 import  pylab
  7 import matplotlib.pyplot as plt
  8 import numpy as np
  9 from matplotlib.animation import FuncAnimation
 10  
 11 ph=r"d:\Users\Administrator\Documents\Kangaroo_test\lin.txt"
 12 mtxt=''
 13 with open(ph,'r') as f:
 14     mtxt=f.read()
 15 txt=mtxt.split('\n')
 16 xn=[]
 17 for i in txt:
 18     if len(i)==0:
 19         continue
 20     sp,ep=i.split(r',')
 21     sp=float(sp)
 22     ep=float(ep)
 23     xn.append(sp+ep*1j)
 24     
 25 # xn=np.linspace(0.2, 3, 128)
 26 # X = fft([i+1/i*1j for i in xn])
 27 
 28 X = fft(xn)
 29 
 30 frames = len(X)  # number of animation frames
 31 n = frames       # number of ciecles
 32 coef = np.arange(frames)
 33 
 34 factor = []
 35 for i in range(len(coef)):
 36   factor.append(X[i] * np.exp(2 * np.pi * 1j * coef[i] * np.arange(frames) / frames) / frames)
 37 pos = [[0 for i in range(frames)]]
 38 for i in range(len(coef)):
 39   pos.append(pos[-1] + factor[i])
 40  
 41 fig, ax = pylab.subplots()
 42 lines = []
 43 circles = []
 44 def _init_lines():  
 45     global  lines  
 46     lines += [ax.plot([], [], alpha=1)[0] for i in range(n)]
 47  
 48 def _init_circles():
 49     global  circles,ax
 50     for i in range(n):
 51         radius = np.abs(pos[i+1][0] - pos[i][0])
 52         circle = plt.Circle((-1, -1), radius, alpha=0.6, fill=False)
 53         circles.append(circle)
 54         ax.add_artist(circle)
 55     ####
 56     circle_cyl=plt.Circle((-1, -1), 0.01, alpha=0.99, fill=True)
 57     circles.append(circle_cyl)
 58     ax.add_artist(circle_cyl)
 59     
 60     
 61 path_x = []
 62 path_y = []
 63  
 64 flag=0 
 65 def init_frame():
 66     global  flag
 67     flag+=1
 68     print ( flag )
 69     ax.clear()
 70     viewZoom=1
 71     ax.set(xlim=[-1*viewZoom, 1*viewZoom], ylim=[-1*viewZoom, 1*viewZoom])
 72     ax.axis('off')
 73     _init_lines()
 74     _init_circles()
 75     global path
 76     path = ax.plot([], [])[0]
 77  
 78 def _render_lines(k):
 79     for i in range(n):
 80         p, q = pos[i][k], pos[i+1][k]
 81         lines[i].set_data([p.real, q.real], [p.imag, q.imag])
 82     ###
 83     circles[-1].center = (q.real, q.imag)
 84  
 85 def _render_circles(k):
 86     for i in range(n):
 87         p = pos[i][k]        
 88         circles[i].center = (p.real, p.imag)
 89     
 90  
 91 def render_frame(k):    
 92     _render_lines(k)
 93     _render_circles(k)
 94     p = pos[n][k]
 95     path_x.append(p.real)
 96     path_y.append(p.imag)
 97     path.set_data(path_x, path_y)
 98     
 99 
100 fq= 1000/120
101 animation = FuncAnimation(fig,
102                           render_frame,
103                           frames=range(frames),
104                           interval=fq,
105                           repeat_delay=fq,
106                           init_func=init_frame,
107                           repeat=1)
108 pylab.show()
109 
110     

 

测试 txt点集文件

-0.049232844231,0.0992831516301
-0.0496022321778,0.057524373699
-0.0521786560248,0.0158532228812
-0.068789539469,0.00963270461548
-0.0977475272242,0.0397213672635
-0.127524661025,0.0690008407628
-0.157906922277,0.0976524479613
-0.18869270775,0.12587051679
-0.219689491178,0.153856795101
-0.224307764655,0.120997161149
-0.225060105945,0.0792422195054
-0.225812447236,0.0374872778613
-0.226564788527,-0.00426766378267
-0.227317129818,-0.0460226054267
-0.228069471109,-0.0877775470707
-0.2288218124,-0.129532488715
-0.229574153691,-0.171287430359
-0.230326494982,-0.213042372003
-0.231078836272,-0.254797313647
-0.231831177563,-0.296552255291
-0.232583518854,-0.338307196935
-0.233335860145,-0.380062138579
-0.234088201436,-0.421817080223
-0.234840542727,-0.463572021867
-0.23406682897,-0.50528613123
-0.236880672834,-0.546783712117
-0.263623481405,-0.541137647928
-0.281501103017,-0.503396002302
-0.29937872463,-0.465654356676
-0.301276260762,-0.430147827424
-0.290720562447,-0.389967229018
-0.287896876373,-0.348316887852
-0.287167934456,-0.306562834091
-0.28618557767,-0.264812670687
-0.285203220884,-0.223062507283
-0.284220864098,-0.181312343879
-0.283238507312,-0.139562180475
-0.282256150526,-0.0978120170714
-0.28127379374,-0.0560618536674
-0.280291436954,-0.0143116902634
-0.279309080168,0.0274384731405
-0.278326723382,0.0691886365445
-0.277344366596,0.110938799948
-0.27636200981,0.152688963352
-0.287757631673,0.146552932914
-0.307708741553,0.10986582613
-0.32837792291,0.0735784356736
-0.349762453472,0.0377080150188
-0.371858329362,0.00227155119055
-0.394660550348,-0.0327147763075
-0.418162605899,-0.0672348579366
-0.442356872926,-0.101273355528
-0.467234546841,-0.134815624914
-0.492785630149,-0.16784772941
-0.518999136371,-0.200356711936
-0.545862957331,-0.232330409182
-0.573364005207,-0.263757612789
-0.601488444217,-0.294628287112
-0.631048719569,-0.314645639541
-0.634556599816,-0.286484294791
-0.607600462821,-0.254588843457
-0.581434248834,-0.222042183578
-0.556048464678,-0.188883244674
-0.531430869407,-0.155149704265
-0.507567456896,-0.12087858525
-0.484442119061,-0.0861051445069
-0.462037250878,-0.0508630768635
-0.440334113567,-0.0151845071957
-0.419313019304,0.0209002208026
-0.398953680235,0.0573624128243
-0.379235309509,0.0941752393255
-0.360136936134,0.131313505097
-0.341637477218,0.16875379103
-0.323715962529,0.20647424773
-0.306351701784,0.244454416099
-0.303080639959,0.27216969355
-0.344297489058,0.265446347935
-0.385433726488,0.25824800967
-0.426385016882,0.250071629365
-0.466646351195,0.239088458028
-0.500376783807,0.250928123094
-0.532084644641,0.278106289523
-0.557011050704,0.303071103797
-0.515408527655,0.306708632128
-0.473864930648,0.310967734644
-0.43237779759,0.31574665731
-0.390945793209,0.320982378234
-0.349568416114,0.326633908111
-0.308246005095,0.332674005145
-0.275977030278,0.346765863936
-0.275977030278,0.388527582879
-0.275977030278,0.430289301823
-0.275977030278,0.472051020766
-0.275977030278,0.513812739709
-0.275977030278,0.555574458653
-0.274585061385,0.597295640806
-0.276287494939,0.638996464807
-0.28380957414,0.680029687615
-0.298748152592,0.718944440366
-0.297625382351,0.749216706773
-0.26445163379,0.723848546108
-0.231277885229,0.698480385444
-0.203246954404,0.674486673854
-0.221631295903,0.638945814015
-0.225088516339,0.59736723647
-0.224180733271,0.555624759081
-0.223651448882,0.513879137779
-0.223651448882,0.472117418835
-0.223651448882,0.430355699892
-0.223651448882,0.388593980948
-0.223651448882,0.346832262005
-0.18564482762,0.348936649945
-0.144519854358,0.356182199128
-0.104087521021,0.366517903668
-0.0692729471465,0.367962386462
-0.0350604139323,0.344013613212
0.00445577023738,0.341487072628
0.0459705228884,0.346020637648
0.0874950226481,0.350465435819
0.128999050686,0.355097138148
0.170467591851,0.36003610983
0.178377667108,0.395564070291
0.179421383971,0.437312744811
0.180465100834,0.479061419332
0.181508817697,0.520810093852
0.18255253456,0.562558768373
0.183966475911,0.60429175821
0.184631178645,0.646037965299
0.181137720451,0.687624866652
0.170481316598,0.72790635113
0.149028237553,0.762660512484
0.180839589147,0.757818732672
0.214013337708,0.732450572008
0.247187086269,0.707082411343
0.243573188398,0.67636623326
0.235023750768,0.635633713983
0.234741398521,0.59390838732
0.234827308111,0.552192822
0.233667707653,0.510447205513
0.232508107195,0.468701589026
0.231348506737,0.426955972539
0.230188906279,0.385210356052
0.258673896229,0.37633261733
0.300056973508,0.38194319334
0.341391180586,0.387901835963
0.382614699534,0.394577838952
0.423459794805,0.403216947781
0.459172135197,0.40280942253
0.492848993988,0.378113059416
0.52652585278,0.353416696303
0.497417022909,0.348068393529
0.455781212146,0.344848148577
0.414223176308,0.34073185564
0.372715368644,0.336135077215
0.331239447846,0.331257855768
0.289783583959,0.326213319875
0.248339490705,0.321072100158
0.254067457036,0.287739446127
0.269259135983,0.248838888364
0.284558925725,0.209980730522
0.299991652879,0.171175157398
0.315590655966,0.132436214085
0.331402601475,0.093783674034
0.347495964708,0.0552474156668
0.363978263373,0.0168762483552
0.381036631681,-0.0212420018369
0.399096145003,-0.0588928395245
0.419485291764,-0.0953381239738
0.440917750289,-0.131177779309
0.464220407337,-0.165822041994
0.491252284537,-0.197593868274
0.527339804956,-0.218127467711
0.567278890667,-0.230230553504
0.608197504829,-0.238539607222
0.642627620885,-0.250679250834
0.622996522734,-0.272809871626
0.581234803791,-0.272809871626
0.539473084847,-0.272809871626
0.497711365904,-0.272809871626
0.45594964696,-0.272809871626
0.426500656201,-0.259535502759
0.416546480979,-0.219024507111
0.40292185668,-0.179551691204
0.388220448469,-0.140463543528
0.372014234006,-0.101996705903
0.354975195035,-0.0638693868485
0.338275592217,-0.0255920137165
0.321849783944,0.0128036387221
0.305657129273,0.0512982980069
0.289670210902,0.0898787955416
0.273869230635,0.12853581791
0.258239301254,0.167262306804
0.2427688144,0.206052783687
0.229786640893,0.232281334092
0.229465406405,0.190520850647
0.229144171917,0.148760367203
0.228822937429,0.106999883758
0.228501702941,0.0652394003139
0.228180468453,0.0234789168694
0.227859233965,-0.0182815665751
0.227537999477,-0.0600420500195
0.227216764989,-0.101802533464
0.226895530501,-0.143563016909
0.226574296013,-0.185323500353
0.226253061525,-0.227083983797
0.225931827037,-0.268844467242
0.225610592549,-0.310604950686
0.225289358061,-0.352365434131
0.224968123573,-0.394125917575
0.224646889085,-0.43588640102
0.224325654597,-0.477646884464
0.22401876692,-0.519407436937
0.221584547955,-0.561059336029
0.19906468746,-0.5577052113
0.176377580266,-0.522643318364
0.153690473072,-0.487581425427
0.156860234921,-0.454667364926
0.170625243215,-0.415520134915
0.17303146128,-0.373896986135
0.171944802191,-0.332185386568
0.172414005371,-0.290426303514
0.172883208552,-0.248667220461
0.173352411732,-0.206908137407
0.173821614913,-0.165149054353
0.174290818093,-0.123389971299
0.174760021273,-0.0816308882456
0.175229224454,-0.0398718051919
0.175698427634,0.00188727786185
0.176167630815,0.0436463609156
0.176636833995,0.0854054439693
0.177106037175,0.127164527023
0.175207597851,0.158036780212
0.157791759241,0.120080803301
0.139487259007,0.0825454333286
0.120282464102,0.0454626820568
0.100167748065,0.00886548443653
0.079135924511,-0.0272121685761
0.0571819701317,-0.0627363416486
0.034303808625,-0.0976724143493
0.0105021376361,-0.131985987466
-0.0142191907165,-0.165642850631
-0.0398533927525,-0.198609781346
-0.0663904076162,-0.230854482667
-0.0938169117738,-0.26234599052
-0.122116493209,-0.293055170007
-0.151269855982,-0.322955120808
-0.178926865678,-0.331175816199
-0.160579153631,-0.298872831607
-0.134272076816,-0.266440128396
-0.108830808941,-0.233323919966
-0.0842698089707,-0.199549773486
-0.0605999511103,-0.165145122392
-0.0378286601426,-0.130139229087
-0.0159597203238,-0.0945625789711
0.00500660487594,-0.058446605189
0.0250735196126,-0.0218232127255
0.0442474576332,0.0152754650861
0.0625380255702,0.0528177239553
0.0799574435883,0.0907719915327
0.0965205884669,0.129107756396
0.112244492731,0.167795387013
0.127148063719,0.206806348429
0.141251949451,0.246113812629
0.154577811451,0.285691571557
0.136344377341,0.3006460966
0.0949585882514,0.295111774587
0.0542692499376,0.285835348353
0.0174923553402,0.280589908553
-0.0154723501092,0.306229123902
-0.0544066540584,0.31337194009
-0.0959340718555,0.309026886034
-0.137185643957,0.302536512272
-0.178219616031,0.29477925772
-0.21911126664,0.286300342756
-0.223651448882,0.248210729329
-0.223651448882,0.206449010385
-0.194387931168,0.181952281736
-0.158034744977,0.161398067816
-0.121812668148,0.14061346029
-0.0855984679436,0.119815071014

  

 

标签:circles,fft,pos,备忘,range,ax,frames,path,numpy
来源: https://www.cnblogs.com/chenyalin/p/15851765.html