PageRenderTime 28ms CodeModel.GetById 22ms RepoModel.GetById 0ms app.codeStats 0ms

/006_test_python_d2o_tp7_projects_linux/webservice/d2o/doc/html/_modules/d2o/analyze/fourier.html

https://gitlab.com/sang.nguyen/practice_git
HTML | 283 lines | 166 code | 116 blank | 1 comment | 0 complexity | 8b4ce2a308bfcd9d23e3697eb230be31 MD5 | raw file
  1. <!DOCTYPE html>
  2. <!--[if IE 8]><html class="no-js lt-ie9" lang="en" > <![endif]-->
  3. <!--[if gt IE 8]><!--> <html class="no-js" lang="en" > <!--<![endif]-->
  4. <head>
  5. <meta charset="utf-8">
  6. <meta name="viewport" content="width=device-width, initial-scale=1.0">
  7. <title>d2o.analyze.fourier &mdash; d2o Python Library 1.0 documentation</title>
  8. <link rel="stylesheet" href="../../../_static/css/theme.css" type="text/css" />
  9. <link rel="top" title="d2o Python Library 1.0 documentation" href="../../../index.html"/>
  10. <link rel="up" title="Module code" href="../../index.html"/>
  11. <script src="../../../_static/js/modernizr.min.js"></script>
  12. </head>
  13. <body class="wy-body-for-nav" role="document">
  14. <div class="wy-grid-for-nav">
  15. <nav data-toggle="wy-nav-shift" class="wy-nav-side">
  16. <div class="wy-side-scroll">
  17. <div class="wy-side-nav-search">
  18. <a href="../../../index.html" class="icon icon-home"> d2o Python Library
  19. </a>
  20. <div class="version">
  21. 1.0
  22. </div>
  23. <div role="search">
  24. <form id="rtd-search-form" class="wy-form" action="../../../search.html" method="get">
  25. <input type="text" name="q" placeholder="Search docs" />
  26. <input type="hidden" name="check_keywords" value="yes" />
  27. <input type="hidden" name="area" value="default" />
  28. </form>
  29. </div>
  30. </div>
  31. <div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="main navigation">
  32. <ul class="simple">
  33. </ul>
  34. </div>
  35. </div>
  36. </nav>
  37. <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
  38. <nav class="wy-nav-top" role="navigation" aria-label="top navigation">
  39. <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
  40. <a href="../../../index.html">d2o Python Library</a>
  41. </nav>
  42. <div class="wy-nav-content">
  43. <div class="rst-content">
  44. <div role="navigation" aria-label="breadcrumbs navigation">
  45. <ul class="wy-breadcrumbs">
  46. <li><a href="../../../index.html">Docs</a> &raquo;</li>
  47. <li><a href="../../index.html">Module code</a> &raquo;</li>
  48. <li>d2o.analyze.fourier</li>
  49. <li class="wy-breadcrumbs-aside">
  50. </li>
  51. </ul>
  52. <hr/>
  53. </div>
  54. <div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article">
  55. <div itemprop="articleBody">
  56. <h1>Source code for d2o.analyze.fourier</h1><div class="highlight"><pre>
  57. <span></span><span class="c1"># -*- encoding: utf-8 -*-</span>
  58. <span class="c1">#</span>
  59. <span class="c1"># Copyright (c) 2016 Chronos AS</span>
  60. <span class="c1">#</span>
  61. <span class="c1"># Authors: Fredrik Stormo, Stefan Remman</span>
  62. <span class="c1"># Contact: kjetil.karlsen@chronosit.no</span>
  63. <span class="kn">import</span> <span class="nn">numpy</span> <span class="kn">as</span> <span class="nn">np</span>
  64. <span class="kn">import</span> <span class="nn">pandas</span> <span class="kn">as</span> <span class="nn">pd</span>
  65. <div class="viewcode-block" id="fftdf"><a class="viewcode-back" href="../../../d2o.analyze.html#d2o.analyze.fourier.fftdf">[docs]</a><span class="k">def</span> <span class="nf">fftdf</span><span class="p">(</span><span class="n">dataFrame</span><span class="p">,</span> <span class="n">keys</span><span class="o">=</span><span class="s2">&quot;ALL&quot;</span><span class="p">):</span>
  66. <span class="sd">&quot;&quot;&quot;</span>
  67. <span class="sd"> Does a fast fourier transform on the columns in a dataframe. All columns</span>
  68. <span class="sd"> need to contain floats except index column.</span>
  69. <span class="sd"> Parameters</span>
  70. <span class="sd"> ----------</span>
  71. <span class="sd"> dataFrame : pd.DataFrame</span>
  72. <span class="sd"> keys : string or list, default &quot;ALL&quot;.</span>
  73. <span class="sd"> What keys to itterate over in DataFrame.</span>
  74. <span class="sd"> Returns</span>
  75. <span class="sd"> -------</span>
  76. <span class="sd"> fftDF : pd.DataFrame</span>
  77. <span class="sd"> Data Frame containing the absolute value of the fourier transformation of</span>
  78. <span class="sd"> the columns in dataFrame.</span>
  79. <span class="sd"> &quot;&quot;&quot;</span>
  80. <span class="n">fftDF</span> <span class="o">=</span> <span class="p">{}</span>
  81. <span class="n">length</span> <span class="o">=</span> <span class="n">dataFrame</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span>
  82. <span class="k">if</span> <span class="n">keys</span> <span class="o">==</span> <span class="s2">&quot;ALL&quot;</span><span class="p">:</span>
  83. <span class="n">keys</span> <span class="o">=</span> <span class="n">dataFrame</span><span class="o">.</span><span class="n">columns</span><span class="o">.</span><span class="n">values</span><span class="o">.</span><span class="n">tolist</span><span class="p">()</span>
  84. <span class="k">elif</span> <span class="nb">type</span><span class="p">(</span><span class="n">keys</span><span class="p">)</span> <span class="o">==</span> <span class="nb">str</span><span class="p">:</span>
  85. <span class="n">keys</span> <span class="o">=</span> <span class="p">[</span><span class="n">keys</span><span class="p">]</span>
  86. <span class="k">for</span> <span class="n">key</span> <span class="ow">in</span> <span class="n">keys</span><span class="p">:</span>
  87. <span class="n">transformed</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">fft</span><span class="o">.</span><span class="n">fft</span><span class="p">(</span><span class="n">dataFrame</span><span class="p">[</span><span class="n">key</span><span class="p">]</span><span class="o">.</span><span class="n">values</span><span class="p">)[</span><span class="mi">0</span><span class="p">:</span><span class="n">length</span><span class="o">/</span><span class="mi">2</span><span class="p">]</span>
  88. <span class="n">fftDF</span><span class="p">[</span><span class="n">key</span><span class="p">]</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">abs</span><span class="p">(</span><span class="n">transformed</span><span class="p">)</span>
  89. <span class="k">return</span> <span class="n">pd</span><span class="o">.</span><span class="n">DataFrame</span><span class="o">.</span><span class="n">from_dict</span><span class="p">(</span><span class="n">fftDF</span><span class="p">)</span></div>
  90. <span class="c1">#import of putils need to be worked out diffrently in the test block. does not work now.</span>
  91. <span class="k">if</span> <span class="n">__name__</span> <span class="o">==</span> <span class="s2">&quot;__main__&quot;</span><span class="p">:</span>
  92. <span class="kn">import</span> <span class="nn">matplotlib.pyplot</span> <span class="kn">as</span> <span class="nn">plt</span>
  93. <span class="c1">#import sys</span>
  94. <span class="c1">#sys.path.append(&quot;~/Documents/chronos/clib/utils&quot;)</span>
  95. <span class="c1">#import putils</span>
  96. <span class="c1">#import imp</span>
  97. <span class="c1">#imp.load_source(&quot;putils&quot;, &quot;~/Documents/chronos/clib/utils/putils&quot;)</span>
  98. <span class="c1">#import d2o.utils.putils as putils</span>
  99. <span class="n">hotel</span> <span class="o">=</span> <span class="s2">&quot;HotelBristol&quot;</span>
  100. <span class="n">path</span> <span class="o">=</span> <span class="s2">&quot;~/Documents/chronos/d2o-datasett/&quot;</span> <span class="o">+</span> <span class="n">hotel</span> <span class="o">+</span> <span class="s2">&quot;.csv&quot;</span>
  101. <span class="c1">#testFrame = pd.read_csv(&quot;~/Documents/chronos/d2o-datasett/BritanniaHotel.csv&quot;, sep=&quot;;&quot;)</span>
  102. <span class="n">testFrame</span> <span class="o">=</span> <span class="n">pd</span><span class="o">.</span><span class="n">read_csv</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">sep</span><span class="o">=</span><span class="s2">&quot;;&quot;</span><span class="p">,</span> <span class="n">index_col</span><span class="o">=</span><span class="s2">&quot;Date&quot;</span><span class="p">)</span>
  103. <span class="c1"># print testFrame.columns.values.tolist()</span>
  104. <span class="c1"># testFrameMav = putils.add_mav(testFrame, 10)</span>
  105. <span class="n">testFrameMav</span> <span class="o">=</span> <span class="n">testFrame</span>
  106. <span class="c1"># figure = plt.figure()</span>
  107. <span class="c1"># ax = figure.add_subplot(1, 1 ,1)</span>
  108. <span class="c1">## ax.plot(testFrameMav[&quot;Total_RV_MAV(10)&quot;][365:400].values)</span>
  109. <span class="c1"># ax.plot(testFrameMav[&quot;Total_RV&quot;][365:400].values)</span>
  110. <span class="c1"># figure.show()</span>
  111. <span class="c1">#</span>
  112. <span class="c1"># raw_input(&quot;enter: &quot;)</span>
  113. <span class="c1">#</span>
  114. <span class="n">length</span> <span class="o">=</span> <span class="n">testFrameMav</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span>
  115. <span class="n">index</span> <span class="o">=</span> <span class="n">length</span> <span class="o">-</span> <span class="mi">365</span><span class="o">*</span><span class="mi">2</span>
  116. <span class="c1"># print length, index</span>
  117. <span class="n">testFrameMav</span> <span class="o">=</span> <span class="n">testFrameMav</span><span class="o">.</span><span class="n">ix</span><span class="p">[</span><span class="n">index</span><span class="p">:]</span>
  118. <span class="c1"># print testFrameMav</span>
  119. <span class="n">otherFrame</span> <span class="o">=</span> <span class="n">fftdf</span><span class="p">(</span><span class="n">testFrameMav</span><span class="p">,</span> <span class="n">keys</span><span class="o">=</span><span class="p">[</span><span class="s2">&quot;Total_RV&quot;</span><span class="p">,</span> <span class="s2">&quot;Total_RN&quot;</span><span class="p">])</span>
  120. <span class="k">print</span> <span class="n">otherFrame</span>
  121. <span class="n">fig</span> <span class="o">=</span> <span class="n">plt</span><span class="o">.</span><span class="n">figure</span><span class="p">()</span>
  122. <span class="n">fig</span><span class="o">.</span><span class="n">suptitle</span><span class="p">(</span><span class="n">hotel</span><span class="p">,</span> <span class="n">fontsize</span><span class="o">=</span><span class="mi">20</span><span class="p">)</span>
  123. <span class="n">ax</span> <span class="o">=</span> <span class="n">fig</span><span class="o">.</span><span class="n">add_subplot</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">1</span><span class="p">)</span>
  124. <span class="n">ax</span><span class="o">.</span><span class="n">plot</span><span class="p">(</span><span class="n">otherFrame</span><span class="p">[</span><span class="s2">&quot;Total_RV&quot;</span><span class="p">][:</span><span class="o">-</span><span class="n">otherFrame</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">/</span><span class="mi">2</span><span class="p">],</span> <span class="n">label</span><span class="o">=</span><span class="s2">&quot;Total_RV&quot;</span><span class="p">)</span>
  125. <span class="c1"># ax.plot(otherFrame[&quot;Total_RV_MAV(10)&quot;][:-otherFrame.shape[0]/2], label=&quot;MAV&quot;)</span>
  126. <span class="n">ax</span><span class="o">.</span><span class="n">grid</span><span class="p">(</span><span class="n">b</span><span class="o">=</span><span class="bp">True</span><span class="p">)</span>
  127. <span class="n">ax</span><span class="o">.</span><span class="n">legend</span><span class="p">()</span>
  128. <span class="n">fig</span><span class="o">.</span><span class="n">show</span><span class="p">()</span>
  129. <span class="c1">#testFrame.plot()</span>
  130. <span class="nb">raw_input</span><span class="p">(</span><span class="s2">&quot;something&quot;</span><span class="p">)</span>
  131. </pre></div>
  132. </div>
  133. </div>
  134. <footer>
  135. <hr/>
  136. <div role="contentinfo">
  137. <p>
  138. &copy; Copyright 2016, Chronos AS.
  139. </p>
  140. </div>
  141. Built with <a href="http://sphinx-doc.org/">Sphinx</a> using a <a href="https://github.com/snide/sphinx_rtd_theme">theme</a> provided by <a href="https://readthedocs.org">Read the Docs</a>.
  142. </footer>
  143. </div>
  144. </div>
  145. </section>
  146. </div>
  147. <script type="text/javascript">
  148. var DOCUMENTATION_OPTIONS = {
  149. URL_ROOT:'../../../',
  150. VERSION:'1.0',
  151. COLLAPSE_INDEX:false,
  152. FILE_SUFFIX:'.html',
  153. HAS_SOURCE: true
  154. };
  155. </script>
  156. <script type="text/javascript" src="../../../_static/jquery.js"></script>
  157. <script type="text/javascript" src="../../../_static/underscore.js"></script>
  158. <script type="text/javascript" src="../../../_static/doctools.js"></script>
  159. <script type="text/javascript" src="https://cdn.mathjax.org/mathjax/latest/MathJax.js?config=TeX-AMS-MML_HTMLorMML"></script>
  160. <script type="text/javascript" src="../../../_static/js/theme.js"></script>
  161. <script type="text/javascript">
  162. jQuery(function () {
  163. SphinxRtdTheme.StickyNav.enable();
  164. });
  165. </script>
  166. </body>
  167. </html>