mirror of
https://github.com/fjosw/pyerrors.git
synced 2025-03-16 07:10:24 +01:00
Documentation updated
This commit is contained in:
fjosw
parent
7ac3a25f86
commit
b86e3c4a26
1 changed files with 23 additions and 18 deletions
|
|
@ -106,10 +106,15 @@
|
|||
<span class="k">for</span> <span class="n">line</span> <span class="ow">in</span> <span class="n">files</span><span class="p">:</span>
|
||||
<span class="n">cnfg_numbers</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">get_cnfg_number</span><span class="p">(</span><span class="n">line</span><span class="p">))</span>
|
||||
|
||||
<span class="k">if</span> <span class="ow">not</span> <span class="nb">all</span><span class="p">(</span><span class="n">np</span><span class="o">.</span><span class="n">diff</span><span class="p">(</span><span class="n">cnfg_numbers</span><span class="p">)</span> <span class="o">==</span> <span class="n">np</span><span class="o">.</span><span class="n">diff</span><span class="p">(</span><span class="n">cnfg_numbers</span><span class="p">)[</span><span class="mi">0</span><span class="p">]):</span>
|
||||
<span class="n">dc</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">unique</span><span class="p">(</span><span class="n">np</span><span class="o">.</span><span class="n">diff</span><span class="p">(</span><span class="n">cnfg_numbers</span><span class="p">))</span>
|
||||
<span class="k">if</span> <span class="n">np</span><span class="o">.</span><span class="n">any</span><span class="p">(</span><span class="n">dc</span> <span class="o"><</span> <span class="mi">0</span><span class="p">):</span>
|
||||
<span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="s2">"Unsorted files"</span><span class="p">)</span>
|
||||
<span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">dc</span><span class="p">)</span> <span class="o">==</span> <span class="mi">1</span><span class="p">:</span>
|
||||
<span class="n">idx</span> <span class="o">=</span> <span class="nb">range</span><span class="p">(</span><span class="n">cnfg_numbers</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="n">cnfg_numbers</span><span class="p">[</span><span class="o">-</span><span class="mi">1</span><span class="p">]</span> <span class="o">+</span> <span class="n">dc</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="n">dc</span><span class="p">[</span><span class="mi">0</span><span class="p">])</span>
|
||||
<span class="k">else</span><span class="p">:</span>
|
||||
<span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="s1">'Configurations are not evenly spaced.'</span><span class="p">)</span>
|
||||
|
||||
<span class="k">return</span> <span class="n">files</span><span class="p">,</span> <span class="n">cnfg_numbers</span>
|
||||
<span class="k">return</span> <span class="n">files</span><span class="p">,</span> <span class="n">idx</span>
|
||||
|
||||
|
||||
<span class="k">def</span> <span class="nf">read_meson_hd5</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">filestem</span><span class="p">,</span> <span class="n">ens_id</span><span class="p">,</span> <span class="n">meson</span><span class="o">=</span><span class="s1">'meson_0'</span><span class="p">,</span> <span class="n">tree</span><span class="o">=</span><span class="s1">'meson'</span><span class="p">):</span>
|
||||
|
|
@ -132,7 +137,7 @@
|
|||
<span class="sd"> from other modules with similar structures.</span>
|
||||
<span class="sd"> """</span>
|
||||
|
||||
<span class="n">files</span><span class="p">,</span> <span class="n">cnfg_numbers</span> <span class="o">=</span> <span class="n">_get_files</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">filestem</span><span class="p">)</span>
|
||||
<span class="n">files</span><span class="p">,</span> <span class="n">idx</span> <span class="o">=</span> <span class="n">_get_files</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">filestem</span><span class="p">)</span>
|
||||
|
||||
<span class="n">corr_data</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
<span class="n">infos</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
|
|
@ -149,7 +154,7 @@
|
|||
|
||||
<span class="n">l_obs</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
<span class="k">for</span> <span class="n">c</span> <span class="ow">in</span> <span class="n">corr_data</span><span class="o">.</span><span class="n">T</span><span class="p">:</span>
|
||||
<span class="n">l_obs</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">Obs</span><span class="p">([</span><span class="n">c</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">cnfg_numbers</span><span class="p">]))</span>
|
||||
<span class="n">l_obs</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">Obs</span><span class="p">([</span><span class="n">c</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">idx</span><span class="p">]))</span>
|
||||
|
||||
<span class="n">corr</span> <span class="o">=</span> <span class="n">Corr</span><span class="p">(</span><span class="n">l_obs</span><span class="p">)</span>
|
||||
<span class="n">corr</span><span class="o">.</span><span class="n">tag</span> <span class="o">=</span> <span class="sa">r</span><span class="s2">", "</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">infos</span><span class="p">)</span>
|
||||
|
|
@ -169,7 +174,7 @@
|
|||
<span class="sd"> 'C' for the last index changing fastest (16 3x3 matrices),</span>
|
||||
<span class="sd"> """</span>
|
||||
|
||||
<span class="n">files</span><span class="p">,</span> <span class="n">cnfg_numbers</span> <span class="o">=</span> <span class="n">_get_files</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">filestem</span><span class="p">)</span>
|
||||
<span class="n">files</span><span class="p">,</span> <span class="n">idx</span> <span class="o">=</span> <span class="n">_get_files</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">filestem</span><span class="p">)</span>
|
||||
|
||||
<span class="n">mom</span> <span class="o">=</span> <span class="kc">None</span>
|
||||
|
||||
|
|
@ -187,8 +192,8 @@
|
|||
|
||||
<span class="n">matrix</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">empty</span><span class="p">((</span><span class="n">rolled_array</span><span class="o">.</span><span class="n">shape</span><span class="p">[:</span><span class="o">-</span><span class="mi">1</span><span class="p">]),</span> <span class="n">dtype</span><span class="o">=</span><span class="nb">object</span><span class="p">)</span>
|
||||
<span class="k">for</span> <span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span> <span class="ow">in</span> <span class="n">np</span><span class="o">.</span><span class="n">ndindex</span><span class="p">(</span><span class="n">rolled_array</span><span class="o">.</span><span class="n">shape</span><span class="p">[:</span><span class="o">-</span><span class="mi">1</span><span class="p">]):</span>
|
||||
<span class="n">real</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">real</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">cnfg_numbers</span><span class="p">])</span>
|
||||
<span class="n">imag</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">imag</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">cnfg_numbers</span><span class="p">])</span>
|
||||
<span class="n">real</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">real</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">idx</span><span class="p">])</span>
|
||||
<span class="n">imag</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">imag</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">idx</span><span class="p">])</span>
|
||||
<span class="n">matrix</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span> <span class="o">=</span> <span class="n">CObs</span><span class="p">(</span><span class="n">real</span><span class="p">,</span> <span class="n">imag</span><span class="p">)</span>
|
||||
|
||||
<span class="k">return</span> <span class="n">Npr_matrix</span><span class="p">(</span><span class="n">matrix</span><span class="o">.</span><span class="n">swapaxes</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span><span class="o">.</span><span class="n">reshape</span><span class="p">((</span><span class="mi">12</span><span class="p">,</span> <span class="mi">12</span><span class="p">),</span> <span class="n">order</span><span class="o">=</span><span class="n">order</span><span class="p">),</span> <span class="n">mom_in</span><span class="o">=</span><span class="n">mom</span><span class="p">)</span>
|
||||
|
|
@ -207,7 +212,7 @@
|
|||
<span class="sd"> 'C' for the last index changing fastest (16 3x3 matrices),</span>
|
||||
<span class="sd"> """</span>
|
||||
|
||||
<span class="n">files</span><span class="p">,</span> <span class="n">cnfg_numbers</span> <span class="o">=</span> <span class="n">_get_files</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">filestem</span><span class="p">)</span>
|
||||
<span class="n">files</span><span class="p">,</span> <span class="n">idx</span> <span class="o">=</span> <span class="n">_get_files</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">filestem</span><span class="p">)</span>
|
||||
|
||||
<span class="n">mom_in</span> <span class="o">=</span> <span class="kc">None</span>
|
||||
<span class="n">mom_out</span> <span class="o">=</span> <span class="kc">None</span>
|
||||
|
|
@ -237,8 +242,8 @@
|
|||
|
||||
<span class="n">matrix</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">empty</span><span class="p">((</span><span class="n">rolled_array</span><span class="o">.</span><span class="n">shape</span><span class="p">[:</span><span class="o">-</span><span class="mi">1</span><span class="p">]),</span> <span class="n">dtype</span><span class="o">=</span><span class="nb">object</span><span class="p">)</span>
|
||||
<span class="k">for</span> <span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span> <span class="ow">in</span> <span class="n">np</span><span class="o">.</span><span class="n">ndindex</span><span class="p">(</span><span class="n">rolled_array</span><span class="o">.</span><span class="n">shape</span><span class="p">[:</span><span class="o">-</span><span class="mi">1</span><span class="p">]):</span>
|
||||
<span class="n">real</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">real</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">cnfg_numbers</span><span class="p">])</span>
|
||||
<span class="n">imag</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">imag</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">cnfg_numbers</span><span class="p">])</span>
|
||||
<span class="n">real</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">real</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">idx</span><span class="p">])</span>
|
||||
<span class="n">imag</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">imag</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">idx</span><span class="p">])</span>
|
||||
<span class="n">matrix</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span> <span class="o">=</span> <span class="n">CObs</span><span class="p">(</span><span class="n">real</span><span class="p">,</span> <span class="n">imag</span><span class="p">)</span>
|
||||
|
||||
<span class="n">result_dict</span><span class="p">[</span><span class="n">key</span><span class="p">]</span> <span class="o">=</span> <span class="n">Npr_matrix</span><span class="p">(</span><span class="n">matrix</span><span class="o">.</span><span class="n">swapaxes</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span><span class="o">.</span><span class="n">reshape</span><span class="p">((</span><span class="mi">12</span><span class="p">,</span> <span class="mi">12</span><span class="p">),</span> <span class="n">order</span><span class="o">=</span><span class="n">order</span><span class="p">),</span> <span class="n">mom_in</span><span class="o">=</span><span class="n">mom_in</span><span class="p">,</span> <span class="n">mom_out</span><span class="o">=</span><span class="n">mom_out</span><span class="p">)</span>
|
||||
|
|
@ -279,7 +284,7 @@
|
|||
<span class="sd"> from other modules with similar structures.</span>
|
||||
<span class="sd"> """</span>
|
||||
|
||||
<span class="n">files</span><span class="p">,</span> <span class="n">cnfg_numbers</span> <span class="o">=</span> <span class="n">_get_files</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">filestem</span><span class="p">)</span>
|
||||
<span class="n">files</span><span class="p">,</span> <span class="n">idx</span> <span class="o">=</span> <span class="n">_get_files</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">filestem</span><span class="p">)</span>
|
||||
|
||||
<span class="n">corr_data</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
<span class="n">infos</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
|
|
@ -296,7 +301,7 @@
|
|||
|
||||
<span class="n">l_obs</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
<span class="k">for</span> <span class="n">c</span> <span class="ow">in</span> <span class="n">corr_data</span><span class="o">.</span><span class="n">T</span><span class="p">:</span>
|
||||
<span class="n">l_obs</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">Obs</span><span class="p">([</span><span class="n">c</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">cnfg_numbers</span><span class="p">]))</span>
|
||||
<span class="n">l_obs</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">Obs</span><span class="p">([</span><span class="n">c</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">idx</span><span class="p">]))</span>
|
||||
|
||||
<span class="n">corr</span> <span class="o">=</span> <span class="n">Corr</span><span class="p">(</span><span class="n">l_obs</span><span class="p">)</span>
|
||||
<span class="n">corr</span><span class="o">.</span><span class="n">tag</span> <span class="o">=</span> <span class="sa">r</span><span class="s2">", "</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">infos</span><span class="p">)</span>
|
||||
|
|
@ -351,7 +356,7 @@ from other modules with similar structures.</li>
|
|||
<span class="sd"> 'C' for the last index changing fastest (16 3x3 matrices),</span>
|
||||
<span class="sd"> """</span>
|
||||
|
||||
<span class="n">files</span><span class="p">,</span> <span class="n">cnfg_numbers</span> <span class="o">=</span> <span class="n">_get_files</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">filestem</span><span class="p">)</span>
|
||||
<span class="n">files</span><span class="p">,</span> <span class="n">idx</span> <span class="o">=</span> <span class="n">_get_files</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">filestem</span><span class="p">)</span>
|
||||
|
||||
<span class="n">mom</span> <span class="o">=</span> <span class="kc">None</span>
|
||||
|
||||
|
|
@ -369,8 +374,8 @@ from other modules with similar structures.</li>
|
|||
|
||||
<span class="n">matrix</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">empty</span><span class="p">((</span><span class="n">rolled_array</span><span class="o">.</span><span class="n">shape</span><span class="p">[:</span><span class="o">-</span><span class="mi">1</span><span class="p">]),</span> <span class="n">dtype</span><span class="o">=</span><span class="nb">object</span><span class="p">)</span>
|
||||
<span class="k">for</span> <span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span> <span class="ow">in</span> <span class="n">np</span><span class="o">.</span><span class="n">ndindex</span><span class="p">(</span><span class="n">rolled_array</span><span class="o">.</span><span class="n">shape</span><span class="p">[:</span><span class="o">-</span><span class="mi">1</span><span class="p">]):</span>
|
||||
<span class="n">real</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">real</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">cnfg_numbers</span><span class="p">])</span>
|
||||
<span class="n">imag</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">imag</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">cnfg_numbers</span><span class="p">])</span>
|
||||
<span class="n">real</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">real</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">idx</span><span class="p">])</span>
|
||||
<span class="n">imag</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">imag</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">idx</span><span class="p">])</span>
|
||||
<span class="n">matrix</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span> <span class="o">=</span> <span class="n">CObs</span><span class="p">(</span><span class="n">real</span><span class="p">,</span> <span class="n">imag</span><span class="p">)</span>
|
||||
|
||||
<span class="k">return</span> <span class="n">Npr_matrix</span><span class="p">(</span><span class="n">matrix</span><span class="o">.</span><span class="n">swapaxes</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span><span class="o">.</span><span class="n">reshape</span><span class="p">((</span><span class="mi">12</span><span class="p">,</span> <span class="mi">12</span><span class="p">),</span> <span class="n">order</span><span class="o">=</span><span class="n">order</span><span class="p">),</span> <span class="n">mom_in</span><span class="o">=</span><span class="n">mom</span><span class="p">)</span>
|
||||
|
|
@ -416,7 +421,7 @@ from other modules with similar structures.</li>
|
|||
<span class="sd"> 'C' for the last index changing fastest (16 3x3 matrices),</span>
|
||||
<span class="sd"> """</span>
|
||||
|
||||
<span class="n">files</span><span class="p">,</span> <span class="n">cnfg_numbers</span> <span class="o">=</span> <span class="n">_get_files</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">filestem</span><span class="p">)</span>
|
||||
<span class="n">files</span><span class="p">,</span> <span class="n">idx</span> <span class="o">=</span> <span class="n">_get_files</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">filestem</span><span class="p">)</span>
|
||||
|
||||
<span class="n">mom_in</span> <span class="o">=</span> <span class="kc">None</span>
|
||||
<span class="n">mom_out</span> <span class="o">=</span> <span class="kc">None</span>
|
||||
|
|
@ -446,8 +451,8 @@ from other modules with similar structures.</li>
|
|||
|
||||
<span class="n">matrix</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">empty</span><span class="p">((</span><span class="n">rolled_array</span><span class="o">.</span><span class="n">shape</span><span class="p">[:</span><span class="o">-</span><span class="mi">1</span><span class="p">]),</span> <span class="n">dtype</span><span class="o">=</span><span class="nb">object</span><span class="p">)</span>
|
||||
<span class="k">for</span> <span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span> <span class="ow">in</span> <span class="n">np</span><span class="o">.</span><span class="n">ndindex</span><span class="p">(</span><span class="n">rolled_array</span><span class="o">.</span><span class="n">shape</span><span class="p">[:</span><span class="o">-</span><span class="mi">1</span><span class="p">]):</span>
|
||||
<span class="n">real</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">real</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">cnfg_numbers</span><span class="p">])</span>
|
||||
<span class="n">imag</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">imag</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">cnfg_numbers</span><span class="p">])</span>
|
||||
<span class="n">real</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">real</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">idx</span><span class="p">])</span>
|
||||
<span class="n">imag</span> <span class="o">=</span> <span class="n">Obs</span><span class="p">([</span><span class="n">rolled_array</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span><span class="o">.</span><span class="n">imag</span><span class="p">],</span> <span class="p">[</span><span class="n">ens_id</span><span class="p">],</span> <span class="n">idl</span><span class="o">=</span><span class="p">[</span><span class="n">idx</span><span class="p">])</span>
|
||||
<span class="n">matrix</span><span class="p">[</span><span class="n">si</span><span class="p">,</span> <span class="n">sj</span><span class="p">,</span> <span class="n">ci</span><span class="p">,</span> <span class="n">cj</span><span class="p">]</span> <span class="o">=</span> <span class="n">CObs</span><span class="p">(</span><span class="n">real</span><span class="p">,</span> <span class="n">imag</span><span class="p">)</span>
|
||||
|
||||
<span class="n">result_dict</span><span class="p">[</span><span class="n">key</span><span class="p">]</span> <span class="o">=</span> <span class="n">Npr_matrix</span><span class="p">(</span><span class="n">matrix</span><span class="o">.</span><span class="n">swapaxes</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span><span class="o">.</span><span class="n">reshape</span><span class="p">((</span><span class="mi">12</span><span class="p">,</span> <span class="mi">12</span><span class="p">),</span> <span class="n">order</span><span class="o">=</span><span class="n">order</span><span class="p">),</span> <span class="n">mom_in</span><span class="o">=</span><span class="n">mom_in</span><span class="p">,</span> <span class="n">mom_out</span><span class="o">=</span><span class="n">mom_out</span><span class="p">)</span>
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue