mirror of
https://github.com/fjosw/pyerrors.git
synced 2025-05-14 19:43:41 +02:00
Documentation updated
This commit is contained in:
fjosw
parent
b6789a63a9
commit
a0c02df621
3 changed files with 335 additions and 341 deletions
|
|
@ -493,7 +493,7 @@
|
|||
</span><span id="L-397"><a href="#L-397"><span class="linenos">397</span></a>
|
||||
</span><span id="L-398"><a href="#L-398"><span class="linenos">398</span></a>
|
||||
</span><span id="L-399"><a href="#L-399"><span class="linenos">399</span></a><span class="c1"># this is based on Mattia Bruno's implementation at https://github.com/mbruno46/pyobs/blob/master/pyobs/IO/xml.py</span>
|
||||
</span><span id="L-400"><a href="#L-400"><span class="linenos">400</span></a><span class="k">def</span> <span class="nf">import_dobs_string</span><span class="p">(</span><span class="n">content</span><span class="p">,</span> <span class="n">noempty</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">full_output</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">separator_insertion</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>
|
||||
</span><span id="L-400"><a href="#L-400"><span class="linenos">400</span></a><span class="k">def</span> <span class="nf">import_dobs_string</span><span class="p">(</span><span class="n">content</span><span class="p">,</span> <span class="n">full_output</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">separator_insertion</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>
|
||||
</span><span id="L-401"><a href="#L-401"><span class="linenos">401</span></a><span class="w"> </span><span class="sd">"""Import a list of Obs from a string in the Zeuthen dobs format.</span>
|
||||
</span><span id="L-402"><a href="#L-402"><span class="linenos">402</span></a>
|
||||
</span><span id="L-403"><a href="#L-403"><span class="linenos">403</span></a><span class="sd"> Tags are not written or recovered automatically.</span>
|
||||
|
|
@ -502,108 +502,108 @@
|
|||
</span><span id="L-406"><a href="#L-406"><span class="linenos">406</span></a><span class="sd"> ----------</span>
|
||||
</span><span id="L-407"><a href="#L-407"><span class="linenos">407</span></a><span class="sd"> content : str</span>
|
||||
</span><span id="L-408"><a href="#L-408"><span class="linenos">408</span></a><span class="sd"> XML string containing the data</span>
|
||||
</span><span id="L-409"><a href="#L-409"><span class="linenos">409</span></a><span class="sd"> noemtpy : bool</span>
|
||||
</span><span id="L-410"><a href="#L-410"><span class="linenos">410</span></a><span class="sd"> If True, ensembles with no contribution to the Obs are not included.</span>
|
||||
</span><span id="L-411"><a href="#L-411"><span class="linenos">411</span></a><span class="sd"> If False, ensembles are included as written in the file, possibly with vanishing entries.</span>
|
||||
</span><span id="L-412"><a href="#L-412"><span class="linenos">412</span></a><span class="sd"> full_output : bool</span>
|
||||
</span><span id="L-413"><a href="#L-413"><span class="linenos">413</span></a><span class="sd"> If True, a dict containing auxiliary information and the data is returned.</span>
|
||||
</span><span id="L-414"><a href="#L-414"><span class="linenos">414</span></a><span class="sd"> If False, only the data is returned as list.</span>
|
||||
</span><span id="L-415"><a href="#L-415"><span class="linenos">415</span></a><span class="sd"> separatior_insertion: str, int or bool</span>
|
||||
</span><span id="L-416"><a href="#L-416"><span class="linenos">416</span></a><span class="sd"> str: replace all occurences of "separator_insertion" within the replica names</span>
|
||||
</span><span id="L-417"><a href="#L-417"><span class="linenos">417</span></a><span class="sd"> by "|%s" % (separator_insertion) when constructing the names of the replica.</span>
|
||||
</span><span id="L-418"><a href="#L-418"><span class="linenos">418</span></a><span class="sd"> int: Insert the separator "|" at the position given by separator_insertion.</span>
|
||||
</span><span id="L-419"><a href="#L-419"><span class="linenos">419</span></a><span class="sd"> True (default): separator "|" is inserted after len(ensname), assuming that the</span>
|
||||
</span><span id="L-420"><a href="#L-420"><span class="linenos">420</span></a><span class="sd"> ensemble name is a prefix to the replica name.</span>
|
||||
</span><span id="L-421"><a href="#L-421"><span class="linenos">421</span></a><span class="sd"> None or False: No separator is inserted.</span>
|
||||
</span><span id="L-422"><a href="#L-422"><span class="linenos">422</span></a>
|
||||
</span><span id="L-423"><a href="#L-423"><span class="linenos">423</span></a><span class="sd"> Returns</span>
|
||||
</span><span id="L-424"><a href="#L-424"><span class="linenos">424</span></a><span class="sd"> -------</span>
|
||||
</span><span id="L-425"><a href="#L-425"><span class="linenos">425</span></a><span class="sd"> res : list[Obs]</span>
|
||||
</span><span id="L-426"><a href="#L-426"><span class="linenos">426</span></a><span class="sd"> Imported data</span>
|
||||
</span><span id="L-427"><a href="#L-427"><span class="linenos">427</span></a><span class="sd"> or</span>
|
||||
</span><span id="L-428"><a href="#L-428"><span class="linenos">428</span></a><span class="sd"> res : dict</span>
|
||||
</span><span id="L-429"><a href="#L-429"><span class="linenos">429</span></a><span class="sd"> Imported data and meta-data</span>
|
||||
</span><span id="L-430"><a href="#L-430"><span class="linenos">430</span></a><span class="sd"> """</span>
|
||||
</span><span id="L-431"><a href="#L-431"><span class="linenos">431</span></a>
|
||||
</span><span id="L-432"><a href="#L-432"><span class="linenos">432</span></a> <span class="n">root</span> <span class="o">=</span> <span class="n">et</span><span class="o">.</span><span class="n">fromstring</span><span class="p">(</span><span class="n">content</span><span class="p">)</span>
|
||||
</span><span id="L-433"><a href="#L-433"><span class="linenos">433</span></a>
|
||||
</span><span id="L-434"><a href="#L-434"><span class="linenos">434</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'OBSERVABLES'</span><span class="p">)</span>
|
||||
</span><span id="L-435"><a href="#L-435"><span class="linenos">435</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'SCHEMA'</span><span class="p">)</span>
|
||||
</span><span id="L-436"><a href="#L-436"><span class="linenos">436</span></a> <span class="n">version</span> <span class="o">=</span> <span class="n">root</span><span class="p">[</span><span class="mi">0</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="L-409"><a href="#L-409"><span class="linenos">409</span></a><span class="sd"> full_output : bool</span>
|
||||
</span><span id="L-410"><a href="#L-410"><span class="linenos">410</span></a><span class="sd"> If True, a dict containing auxiliary information and the data is returned.</span>
|
||||
</span><span id="L-411"><a href="#L-411"><span class="linenos">411</span></a><span class="sd"> If False, only the data is returned as list.</span>
|
||||
</span><span id="L-412"><a href="#L-412"><span class="linenos">412</span></a><span class="sd"> separatior_insertion: str, int or bool</span>
|
||||
</span><span id="L-413"><a href="#L-413"><span class="linenos">413</span></a><span class="sd"> str: replace all occurences of "separator_insertion" within the replica names</span>
|
||||
</span><span id="L-414"><a href="#L-414"><span class="linenos">414</span></a><span class="sd"> by "|%s" % (separator_insertion) when constructing the names of the replica.</span>
|
||||
</span><span id="L-415"><a href="#L-415"><span class="linenos">415</span></a><span class="sd"> int: Insert the separator "|" at the position given by separator_insertion.</span>
|
||||
</span><span id="L-416"><a href="#L-416"><span class="linenos">416</span></a><span class="sd"> True (default): separator "|" is inserted after len(ensname), assuming that the</span>
|
||||
</span><span id="L-417"><a href="#L-417"><span class="linenos">417</span></a><span class="sd"> ensemble name is a prefix to the replica name.</span>
|
||||
</span><span id="L-418"><a href="#L-418"><span class="linenos">418</span></a><span class="sd"> None or False: No separator is inserted.</span>
|
||||
</span><span id="L-419"><a href="#L-419"><span class="linenos">419</span></a>
|
||||
</span><span id="L-420"><a href="#L-420"><span class="linenos">420</span></a><span class="sd"> Returns</span>
|
||||
</span><span id="L-421"><a href="#L-421"><span class="linenos">421</span></a><span class="sd"> -------</span>
|
||||
</span><span id="L-422"><a href="#L-422"><span class="linenos">422</span></a><span class="sd"> res : list[Obs]</span>
|
||||
</span><span id="L-423"><a href="#L-423"><span class="linenos">423</span></a><span class="sd"> Imported data</span>
|
||||
</span><span id="L-424"><a href="#L-424"><span class="linenos">424</span></a><span class="sd"> or</span>
|
||||
</span><span id="L-425"><a href="#L-425"><span class="linenos">425</span></a><span class="sd"> res : dict</span>
|
||||
</span><span id="L-426"><a href="#L-426"><span class="linenos">426</span></a><span class="sd"> Imported data and meta-data</span>
|
||||
</span><span id="L-427"><a href="#L-427"><span class="linenos">427</span></a><span class="sd"> """</span>
|
||||
</span><span id="L-428"><a href="#L-428"><span class="linenos">428</span></a>
|
||||
</span><span id="L-429"><a href="#L-429"><span class="linenos">429</span></a> <span class="n">root</span> <span class="o">=</span> <span class="n">et</span><span class="o">.</span><span class="n">fromstring</span><span class="p">(</span><span class="n">content</span><span class="p">)</span>
|
||||
</span><span id="L-430"><a href="#L-430"><span class="linenos">430</span></a>
|
||||
</span><span id="L-431"><a href="#L-431"><span class="linenos">431</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'OBSERVABLES'</span><span class="p">)</span>
|
||||
</span><span id="L-432"><a href="#L-432"><span class="linenos">432</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'SCHEMA'</span><span class="p">)</span>
|
||||
</span><span id="L-433"><a href="#L-433"><span class="linenos">433</span></a> <span class="n">version</span> <span class="o">=</span> <span class="n">root</span><span class="p">[</span><span class="mi">0</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="L-434"><a href="#L-434"><span class="linenos">434</span></a>
|
||||
</span><span id="L-435"><a href="#L-435"><span class="linenos">435</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'origin'</span><span class="p">)</span>
|
||||
</span><span id="L-436"><a href="#L-436"><span class="linenos">436</span></a> <span class="n">file_origin</span> <span class="o">=</span> <span class="n">_etree_to_dict</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">1</span><span class="p">])[</span><span class="s1">'origin'</span><span class="p">]</span>
|
||||
</span><span id="L-437"><a href="#L-437"><span class="linenos">437</span></a>
|
||||
</span><span id="L-438"><a href="#L-438"><span class="linenos">438</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'origin'</span><span class="p">)</span>
|
||||
</span><span id="L-439"><a href="#L-439"><span class="linenos">439</span></a> <span class="n">file_origin</span> <span class="o">=</span> <span class="n">_etree_to_dict</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">1</span><span class="p">])[</span><span class="s1">'origin'</span><span class="p">]</span>
|
||||
</span><span id="L-440"><a href="#L-440"><span class="linenos">440</span></a>
|
||||
</span><span id="L-441"><a href="#L-441"><span class="linenos">441</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'dobs'</span><span class="p">)</span>
|
||||
</span><span id="L-442"><a href="#L-442"><span class="linenos">442</span></a>
|
||||
</span><span id="L-443"><a href="#L-443"><span class="linenos">443</span></a> <span class="n">dobs</span> <span class="o">=</span> <span class="n">root</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span>
|
||||
</span><span id="L-444"><a href="#L-444"><span class="linenos">444</span></a>
|
||||
</span><span id="L-445"><a href="#L-445"><span class="linenos">445</span></a> <span class="n">descriptiond</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-446"><a href="#L-446"><span class="linenos">446</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">3</span><span class="p">):</span>
|
||||
</span><span id="L-447"><a href="#L-447"><span class="linenos">447</span></a> <span class="n">descriptiond</span><span class="p">[</span><span class="n">dobs</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span><span class="p">]</span> <span class="o">=</span> <span class="n">dobs</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="L-448"><a href="#L-448"><span class="linenos">448</span></a>
|
||||
</span><span id="L-449"><a href="#L-449"><span class="linenos">449</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'array'</span><span class="p">)</span>
|
||||
</span><span id="L-450"><a href="#L-450"><span class="linenos">450</span></a>
|
||||
</span><span id="L-451"><a href="#L-451"><span class="linenos">451</span></a> <span class="n">symbol</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="L-452"><a href="#L-452"><span class="linenos">452</span></a> <span class="k">if</span> <span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'symbol'</span><span class="p">:</span>
|
||||
</span><span id="L-453"><a href="#L-453"><span class="linenos">453</span></a> <span class="n">symbol</span> <span class="o">=</span> <span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="L-454"><a href="#L-454"><span class="linenos">454</span></a> <span class="n">descriptiond</span><span class="p">[</span><span class="s1">'symbol'</span><span class="p">]</span> <span class="o">=</span> <span class="n">symbol</span>
|
||||
</span><span id="L-455"><a href="#L-455"><span class="linenos">455</span></a> <span class="n">mean</span> <span class="o">=</span> <span class="n">_import_array</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">])[</span><span class="mi">0</span><span class="p">]</span>
|
||||
</span><span id="L-456"><a href="#L-456"><span class="linenos">456</span></a>
|
||||
</span><span id="L-457"><a href="#L-457"><span class="linenos">457</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">4</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"ne"</span><span class="p">)</span>
|
||||
</span><span id="L-458"><a href="#L-458"><span class="linenos">458</span></a> <span class="n">ne</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">4</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">())</span>
|
||||
</span><span id="L-459"><a href="#L-459"><span class="linenos">459</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">5</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"nc"</span><span class="p">)</span>
|
||||
</span><span id="L-460"><a href="#L-460"><span class="linenos">460</span></a> <span class="n">nc</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">5</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">())</span>
|
||||
</span><span id="L-461"><a href="#L-461"><span class="linenos">461</span></a>
|
||||
</span><span id="L-462"><a href="#L-462"><span class="linenos">462</span></a> <span class="n">idld</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-463"><a href="#L-463"><span class="linenos">463</span></a> <span class="n">deltad</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-464"><a href="#L-464"><span class="linenos">464</span></a> <span class="n">covd</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-465"><a href="#L-465"><span class="linenos">465</span></a> <span class="n">gradd</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-466"><a href="#L-466"><span class="linenos">466</span></a> <span class="n">names</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="L-467"><a href="#L-467"><span class="linenos">467</span></a> <span class="n">e_names</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="L-468"><a href="#L-468"><span class="linenos">468</span></a> <span class="n">enstags</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-469"><a href="#L-469"><span class="linenos">469</span></a> <span class="k">for</span> <span class="n">k</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">6</span><span class="p">,</span> <span class="nb">len</span><span class="p">(</span><span class="nb">list</span><span class="p">(</span><span class="n">dobs</span><span class="p">))):</span>
|
||||
</span><span id="L-470"><a href="#L-470"><span class="linenos">470</span></a> <span class="k">if</span> <span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"edata"</span><span class="p">:</span>
|
||||
</span><span id="L-471"><a href="#L-471"><span class="linenos">471</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"enstag"</span><span class="p">)</span>
|
||||
</span><span id="L-472"><a href="#L-472"><span class="linenos">472</span></a> <span class="n">ename</span> <span class="o">=</span> <span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="L-473"><a href="#L-473"><span class="linenos">473</span></a> <span class="n">e_names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">ename</span><span class="p">)</span>
|
||||
</span><span id="L-474"><a href="#L-474"><span class="linenos">474</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"nr"</span><span class="p">)</span>
|
||||
</span><span id="L-475"><a href="#L-475"><span class="linenos">475</span></a> <span class="n">R</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">())</span>
|
||||
</span><span id="L-476"><a href="#L-476"><span class="linenos">476</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">2</span><span class="p">,</span> <span class="mi">2</span> <span class="o">+</span> <span class="n">R</span><span class="p">):</span>
|
||||
</span><span id="L-477"><a href="#L-477"><span class="linenos">477</span></a> <span class="n">deltas</span><span class="p">,</span> <span class="n">rname</span><span class="p">,</span> <span class="n">idx</span> <span class="o">=</span> <span class="n">_import_rdata</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="n">i</span><span class="p">])</span>
|
||||
</span><span id="L-478"><a href="#L-478"><span class="linenos">478</span></a> <span class="k">if</span> <span class="n">separator_insertion</span> <span class="ow">is</span> <span class="kc">None</span> <span class="ow">or</span> <span class="kc">False</span><span class="p">:</span>
|
||||
</span><span id="L-479"><a href="#L-479"><span class="linenos">479</span></a> <span class="k">pass</span>
|
||||
</span><span id="L-480"><a href="#L-480"><span class="linenos">480</span></a> <span class="k">elif</span> <span class="n">separator_insertion</span> <span class="ow">is</span> <span class="kc">True</span><span class="p">:</span>
|
||||
</span><span id="L-481"><a href="#L-481"><span class="linenos">481</span></a> <span class="k">if</span> <span class="n">rname</span><span class="o">.</span><span class="n">startswith</span><span class="p">(</span><span class="n">ename</span><span class="p">):</span>
|
||||
</span><span id="L-482"><a href="#L-482"><span class="linenos">482</span></a> <span class="n">rname</span> <span class="o">=</span> <span class="n">rname</span><span class="p">[:</span><span class="nb">len</span><span class="p">(</span><span class="n">ename</span><span class="p">)]</span> <span class="o">+</span> <span class="s1">'|'</span> <span class="o">+</span> <span class="n">rname</span><span class="p">[</span><span class="nb">len</span><span class="p">(</span><span class="n">ename</span><span class="p">):]</span>
|
||||
</span><span id="L-483"><a href="#L-483"><span class="linenos">483</span></a> <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">,</span> <span class="nb">int</span><span class="p">):</span>
|
||||
</span><span id="L-484"><a href="#L-484"><span class="linenos">484</span></a> <span class="n">rname</span> <span class="o">=</span> <span class="n">rname</span><span class="p">[:</span><span class="n">separator_insertion</span><span class="p">]</span> <span class="o">+</span> <span class="s1">'|'</span> <span class="o">+</span> <span class="n">rname</span><span class="p">[</span><span class="n">separator_insertion</span><span class="p">:]</span>
|
||||
</span><span id="L-485"><a href="#L-485"><span class="linenos">485</span></a> <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">,</span> <span class="nb">str</span><span class="p">):</span>
|
||||
</span><span id="L-486"><a href="#L-486"><span class="linenos">486</span></a> <span class="n">rname</span> <span class="o">=</span> <span class="n">rname</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">,</span> <span class="s2">"|</span><span class="si">%s</span><span class="s2">"</span> <span class="o">%</span> <span class="p">(</span><span class="n">separator_insertion</span><span class="p">))</span>
|
||||
</span><span id="L-438"><a href="#L-438"><span class="linenos">438</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'dobs'</span><span class="p">)</span>
|
||||
</span><span id="L-439"><a href="#L-439"><span class="linenos">439</span></a>
|
||||
</span><span id="L-440"><a href="#L-440"><span class="linenos">440</span></a> <span class="n">dobs</span> <span class="o">=</span> <span class="n">root</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span>
|
||||
</span><span id="L-441"><a href="#L-441"><span class="linenos">441</span></a>
|
||||
</span><span id="L-442"><a href="#L-442"><span class="linenos">442</span></a> <span class="n">descriptiond</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-443"><a href="#L-443"><span class="linenos">443</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">3</span><span class="p">):</span>
|
||||
</span><span id="L-444"><a href="#L-444"><span class="linenos">444</span></a> <span class="n">descriptiond</span><span class="p">[</span><span class="n">dobs</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span><span class="p">]</span> <span class="o">=</span> <span class="n">dobs</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="L-445"><a href="#L-445"><span class="linenos">445</span></a>
|
||||
</span><span id="L-446"><a href="#L-446"><span class="linenos">446</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'array'</span><span class="p">)</span>
|
||||
</span><span id="L-447"><a href="#L-447"><span class="linenos">447</span></a>
|
||||
</span><span id="L-448"><a href="#L-448"><span class="linenos">448</span></a> <span class="n">symbol</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="L-449"><a href="#L-449"><span class="linenos">449</span></a> <span class="k">if</span> <span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'symbol'</span><span class="p">:</span>
|
||||
</span><span id="L-450"><a href="#L-450"><span class="linenos">450</span></a> <span class="n">symbol</span> <span class="o">=</span> <span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="L-451"><a href="#L-451"><span class="linenos">451</span></a> <span class="n">descriptiond</span><span class="p">[</span><span class="s1">'symbol'</span><span class="p">]</span> <span class="o">=</span> <span class="n">symbol</span>
|
||||
</span><span id="L-452"><a href="#L-452"><span class="linenos">452</span></a> <span class="n">mean</span> <span class="o">=</span> <span class="n">_import_array</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">])[</span><span class="mi">0</span><span class="p">]</span>
|
||||
</span><span id="L-453"><a href="#L-453"><span class="linenos">453</span></a>
|
||||
</span><span id="L-454"><a href="#L-454"><span class="linenos">454</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">4</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"ne"</span><span class="p">)</span>
|
||||
</span><span id="L-455"><a href="#L-455"><span class="linenos">455</span></a> <span class="n">ne</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">4</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">())</span>
|
||||
</span><span id="L-456"><a href="#L-456"><span class="linenos">456</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">5</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"nc"</span><span class="p">)</span>
|
||||
</span><span id="L-457"><a href="#L-457"><span class="linenos">457</span></a>
|
||||
</span><span id="L-458"><a href="#L-458"><span class="linenos">458</span></a> <span class="n">idld</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-459"><a href="#L-459"><span class="linenos">459</span></a> <span class="n">deltad</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-460"><a href="#L-460"><span class="linenos">460</span></a> <span class="n">covd</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-461"><a href="#L-461"><span class="linenos">461</span></a> <span class="n">gradd</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-462"><a href="#L-462"><span class="linenos">462</span></a> <span class="n">names</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="L-463"><a href="#L-463"><span class="linenos">463</span></a> <span class="n">e_names</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="L-464"><a href="#L-464"><span class="linenos">464</span></a> <span class="n">enstags</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-465"><a href="#L-465"><span class="linenos">465</span></a> <span class="k">for</span> <span class="n">k</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">6</span><span class="p">,</span> <span class="nb">len</span><span class="p">(</span><span class="nb">list</span><span class="p">(</span><span class="n">dobs</span><span class="p">))):</span>
|
||||
</span><span id="L-466"><a href="#L-466"><span class="linenos">466</span></a> <span class="k">if</span> <span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"edata"</span><span class="p">:</span>
|
||||
</span><span id="L-467"><a href="#L-467"><span class="linenos">467</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"enstag"</span><span class="p">)</span>
|
||||
</span><span id="L-468"><a href="#L-468"><span class="linenos">468</span></a> <span class="n">ename</span> <span class="o">=</span> <span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="L-469"><a href="#L-469"><span class="linenos">469</span></a> <span class="n">e_names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">ename</span><span class="p">)</span>
|
||||
</span><span id="L-470"><a href="#L-470"><span class="linenos">470</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"nr"</span><span class="p">)</span>
|
||||
</span><span id="L-471"><a href="#L-471"><span class="linenos">471</span></a> <span class="n">R</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">())</span>
|
||||
</span><span id="L-472"><a href="#L-472"><span class="linenos">472</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">2</span><span class="p">,</span> <span class="mi">2</span> <span class="o">+</span> <span class="n">R</span><span class="p">):</span>
|
||||
</span><span id="L-473"><a href="#L-473"><span class="linenos">473</span></a> <span class="n">deltas</span><span class="p">,</span> <span class="n">rname</span><span class="p">,</span> <span class="n">idx</span> <span class="o">=</span> <span class="n">_import_rdata</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="n">i</span><span class="p">])</span>
|
||||
</span><span id="L-474"><a href="#L-474"><span class="linenos">474</span></a> <span class="k">if</span> <span class="n">separator_insertion</span> <span class="ow">is</span> <span class="kc">None</span> <span class="ow">or</span> <span class="kc">False</span><span class="p">:</span>
|
||||
</span><span id="L-475"><a href="#L-475"><span class="linenos">475</span></a> <span class="k">pass</span>
|
||||
</span><span id="L-476"><a href="#L-476"><span class="linenos">476</span></a> <span class="k">elif</span> <span class="n">separator_insertion</span> <span class="ow">is</span> <span class="kc">True</span><span class="p">:</span>
|
||||
</span><span id="L-477"><a href="#L-477"><span class="linenos">477</span></a> <span class="k">if</span> <span class="n">rname</span><span class="o">.</span><span class="n">startswith</span><span class="p">(</span><span class="n">ename</span><span class="p">):</span>
|
||||
</span><span id="L-478"><a href="#L-478"><span class="linenos">478</span></a> <span class="n">rname</span> <span class="o">=</span> <span class="n">rname</span><span class="p">[:</span><span class="nb">len</span><span class="p">(</span><span class="n">ename</span><span class="p">)]</span> <span class="o">+</span> <span class="s1">'|'</span> <span class="o">+</span> <span class="n">rname</span><span class="p">[</span><span class="nb">len</span><span class="p">(</span><span class="n">ename</span><span class="p">):]</span>
|
||||
</span><span id="L-479"><a href="#L-479"><span class="linenos">479</span></a> <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">,</span> <span class="nb">int</span><span class="p">):</span>
|
||||
</span><span id="L-480"><a href="#L-480"><span class="linenos">480</span></a> <span class="n">rname</span> <span class="o">=</span> <span class="n">rname</span><span class="p">[:</span><span class="n">separator_insertion</span><span class="p">]</span> <span class="o">+</span> <span class="s1">'|'</span> <span class="o">+</span> <span class="n">rname</span><span class="p">[</span><span class="n">separator_insertion</span><span class="p">:]</span>
|
||||
</span><span id="L-481"><a href="#L-481"><span class="linenos">481</span></a> <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">,</span> <span class="nb">str</span><span class="p">):</span>
|
||||
</span><span id="L-482"><a href="#L-482"><span class="linenos">482</span></a> <span class="n">rname</span> <span class="o">=</span> <span class="n">rname</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">,</span> <span class="s2">"|</span><span class="si">%s</span><span class="s2">"</span> <span class="o">%</span> <span class="p">(</span><span class="n">separator_insertion</span><span class="p">))</span>
|
||||
</span><span id="L-483"><a href="#L-483"><span class="linenos">483</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="L-484"><a href="#L-484"><span class="linenos">484</span></a> <span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="s2">"separator_insertion has to be string or int, is "</span><span class="p">,</span> <span class="nb">type</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">))</span>
|
||||
</span><span id="L-485"><a href="#L-485"><span class="linenos">485</span></a> <span class="k">if</span> <span class="s1">'|'</span> <span class="ow">in</span> <span class="n">rname</span><span class="p">:</span>
|
||||
</span><span id="L-486"><a href="#L-486"><span class="linenos">486</span></a> <span class="n">new_ename</span> <span class="o">=</span> <span class="n">rname</span><span class="p">[:</span><span class="n">rname</span><span class="o">.</span><span class="n">index</span><span class="p">(</span><span class="s1">'|'</span><span class="p">)]</span>
|
||||
</span><span id="L-487"><a href="#L-487"><span class="linenos">487</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="L-488"><a href="#L-488"><span class="linenos">488</span></a> <span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="s2">"separator_insertion has to be string or int, is "</span><span class="p">,</span> <span class="nb">type</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">))</span>
|
||||
</span><span id="L-489"><a href="#L-489"><span class="linenos">489</span></a> <span class="k">if</span> <span class="s1">'|'</span> <span class="ow">in</span> <span class="n">rname</span><span class="p">:</span>
|
||||
</span><span id="L-490"><a href="#L-490"><span class="linenos">490</span></a> <span class="n">new_ename</span> <span class="o">=</span> <span class="n">rname</span><span class="p">[:</span><span class="n">rname</span><span class="o">.</span><span class="n">index</span><span class="p">(</span><span class="s1">'|'</span><span class="p">)]</span>
|
||||
</span><span id="L-491"><a href="#L-491"><span class="linenos">491</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="L-492"><a href="#L-492"><span class="linenos">492</span></a> <span class="n">new_ename</span> <span class="o">=</span> <span class="n">ename</span>
|
||||
</span><span id="L-493"><a href="#L-493"><span class="linenos">493</span></a> <span class="n">enstags</span><span class="p">[</span><span class="n">new_ename</span><span class="p">]</span> <span class="o">=</span> <span class="n">ename</span>
|
||||
</span><span id="L-494"><a href="#L-494"><span class="linenos">494</span></a> <span class="n">idld</span><span class="p">[</span><span class="n">rname</span><span class="p">]</span> <span class="o">=</span> <span class="n">idx</span>
|
||||
</span><span id="L-495"><a href="#L-495"><span class="linenos">495</span></a> <span class="n">deltad</span><span class="p">[</span><span class="n">rname</span><span class="p">]</span> <span class="o">=</span> <span class="n">deltas</span>
|
||||
</span><span id="L-496"><a href="#L-496"><span class="linenos">496</span></a> <span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">rname</span><span class="p">)</span>
|
||||
</span><span id="L-497"><a href="#L-497"><span class="linenos">497</span></a> <span class="k">elif</span> <span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"cdata"</span><span class="p">:</span>
|
||||
</span><span id="L-498"><a href="#L-498"><span class="linenos">498</span></a> <span class="n">cname</span><span class="p">,</span> <span class="n">cov</span><span class="p">,</span> <span class="n">grad</span> <span class="o">=</span> <span class="n">_import_cdata</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">])</span>
|
||||
</span><span id="L-499"><a href="#L-499"><span class="linenos">499</span></a> <span class="n">covd</span><span class="p">[</span><span class="n">cname</span><span class="p">]</span> <span class="o">=</span> <span class="n">cov</span>
|
||||
</span><span id="L-500"><a href="#L-500"><span class="linenos">500</span></a> <span class="k">if</span> <span class="n">grad</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span> <span class="o">==</span> <span class="mi">1</span><span class="p">:</span>
|
||||
</span><span id="L-501"><a href="#L-501"><span class="linenos">501</span></a> <span class="n">gradd</span><span class="p">[</span><span class="n">cname</span><span class="p">]</span> <span class="o">=</span> <span class="p">[</span><span class="n">grad</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">mean</span><span class="p">))]</span>
|
||||
</span><span id="L-502"><a href="#L-502"><span class="linenos">502</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="L-503"><a href="#L-503"><span class="linenos">503</span></a> <span class="n">gradd</span><span class="p">[</span><span class="n">cname</span><span class="p">]</span> <span class="o">=</span> <span class="n">grad</span><span class="o">.</span><span class="n">T</span>
|
||||
</span><span id="L-504"><a href="#L-504"><span class="linenos">504</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="L-505"><a href="#L-505"><span class="linenos">505</span></a> <span class="n">_check</span><span class="p">(</span><span class="kc">False</span><span class="p">)</span>
|
||||
</span><span id="L-506"><a href="#L-506"><span class="linenos">506</span></a> <span class="n">names</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="nb">set</span><span class="p">(</span><span class="n">names</span><span class="p">))</span>
|
||||
</span><span id="L-507"><a href="#L-507"><span class="linenos">507</span></a>
|
||||
</span><span id="L-508"><a href="#L-508"><span class="linenos">508</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">names</span><span class="p">:</span>
|
||||
</span><span id="L-509"><a href="#L-509"><span class="linenos">509</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">])):</span>
|
||||
</span><span id="L-510"><a href="#L-510"><span class="linenos">510</span></a> <span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">]</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">array</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])</span> <span class="o">+</span> <span class="n">mean</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
|
||||
</span><span id="L-488"><a href="#L-488"><span class="linenos">488</span></a> <span class="n">new_ename</span> <span class="o">=</span> <span class="n">ename</span>
|
||||
</span><span id="L-489"><a href="#L-489"><span class="linenos">489</span></a> <span class="n">enstags</span><span class="p">[</span><span class="n">new_ename</span><span class="p">]</span> <span class="o">=</span> <span class="n">ename</span>
|
||||
</span><span id="L-490"><a href="#L-490"><span class="linenos">490</span></a> <span class="n">idld</span><span class="p">[</span><span class="n">rname</span><span class="p">]</span> <span class="o">=</span> <span class="n">idx</span>
|
||||
</span><span id="L-491"><a href="#L-491"><span class="linenos">491</span></a> <span class="n">deltad</span><span class="p">[</span><span class="n">rname</span><span class="p">]</span> <span class="o">=</span> <span class="n">deltas</span>
|
||||
</span><span id="L-492"><a href="#L-492"><span class="linenos">492</span></a> <span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">rname</span><span class="p">)</span>
|
||||
</span><span id="L-493"><a href="#L-493"><span class="linenos">493</span></a> <span class="k">elif</span> <span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"cdata"</span><span class="p">:</span>
|
||||
</span><span id="L-494"><a href="#L-494"><span class="linenos">494</span></a> <span class="n">cname</span><span class="p">,</span> <span class="n">cov</span><span class="p">,</span> <span class="n">grad</span> <span class="o">=</span> <span class="n">_import_cdata</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">])</span>
|
||||
</span><span id="L-495"><a href="#L-495"><span class="linenos">495</span></a> <span class="n">covd</span><span class="p">[</span><span class="n">cname</span><span class="p">]</span> <span class="o">=</span> <span class="n">cov</span>
|
||||
</span><span id="L-496"><a href="#L-496"><span class="linenos">496</span></a> <span class="k">if</span> <span class="n">grad</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span> <span class="o">==</span> <span class="mi">1</span><span class="p">:</span>
|
||||
</span><span id="L-497"><a href="#L-497"><span class="linenos">497</span></a> <span class="n">gradd</span><span class="p">[</span><span class="n">cname</span><span class="p">]</span> <span class="o">=</span> <span class="p">[</span><span class="n">grad</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">mean</span><span class="p">))]</span>
|
||||
</span><span id="L-498"><a href="#L-498"><span class="linenos">498</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="L-499"><a href="#L-499"><span class="linenos">499</span></a> <span class="n">gradd</span><span class="p">[</span><span class="n">cname</span><span class="p">]</span> <span class="o">=</span> <span class="n">grad</span><span class="o">.</span><span class="n">T</span>
|
||||
</span><span id="L-500"><a href="#L-500"><span class="linenos">500</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="L-501"><a href="#L-501"><span class="linenos">501</span></a> <span class="n">_check</span><span class="p">(</span><span class="kc">False</span><span class="p">)</span>
|
||||
</span><span id="L-502"><a href="#L-502"><span class="linenos">502</span></a> <span class="n">names</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="nb">set</span><span class="p">(</span><span class="n">names</span><span class="p">))</span>
|
||||
</span><span id="L-503"><a href="#L-503"><span class="linenos">503</span></a>
|
||||
</span><span id="L-504"><a href="#L-504"><span class="linenos">504</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">names</span><span class="p">:</span>
|
||||
</span><span id="L-505"><a href="#L-505"><span class="linenos">505</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">])):</span>
|
||||
</span><span id="L-506"><a href="#L-506"><span class="linenos">506</span></a> <span class="n">tmp</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">zeros_like</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])</span>
|
||||
</span><span id="L-507"><a href="#L-507"><span class="linenos">507</span></a> <span class="k">for</span> <span class="n">j</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])):</span>
|
||||
</span><span id="L-508"><a href="#L-508"><span class="linenos">508</span></a> <span class="k">if</span> <span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">][</span><span class="n">j</span><span class="p">]</span> <span class="o">!=</span> <span class="mf">0.</span><span class="p">:</span>
|
||||
</span><span id="L-509"><a href="#L-509"><span class="linenos">509</span></a> <span class="n">tmp</span><span class="p">[</span><span class="n">j</span><span class="p">]</span> <span class="o">=</span> <span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">][</span><span class="n">j</span><span class="p">]</span> <span class="o">+</span> <span class="n">mean</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
|
||||
</span><span id="L-510"><a href="#L-510"><span class="linenos">510</span></a> <span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">]</span> <span class="o">=</span> <span class="n">tmp</span>
|
||||
</span><span id="L-511"><a href="#L-511"><span class="linenos">511</span></a>
|
||||
</span><span id="L-512"><a href="#L-512"><span class="linenos">512</span></a> <span class="n">res</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="L-513"><a href="#L-513"><span class="linenos">513</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">mean</span><span class="p">)):</span>
|
||||
|
|
@ -612,70 +612,70 @@
|
|||
</span><span id="L-516"><a href="#L-516"><span class="linenos">516</span></a> <span class="n">obs_names</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="L-517"><a href="#L-517"><span class="linenos">517</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">names</span><span class="p">:</span>
|
||||
</span><span id="L-518"><a href="#L-518"><span class="linenos">518</span></a> <span class="n">h</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">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])</span>
|
||||
</span><span id="L-519"><a href="#L-519"><span class="linenos">519</span></a> <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">h</span><span class="p">)</span> <span class="o">==</span> <span class="mi">1</span> <span class="ow">and</span> <span class="n">np</span><span class="o">.</span><span class="n">all</span><span class="p">(</span><span class="n">h</span> <span class="o">==</span> <span class="n">mean</span><span class="p">[</span><span class="n">i</span><span class="p">])</span> <span class="ow">and</span> <span class="n">noempty</span><span class="p">:</span>
|
||||
</span><span id="L-519"><a href="#L-519"><span class="linenos">519</span></a> <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">h</span><span class="p">)</span> <span class="o">==</span> <span class="mi">1</span> <span class="ow">and</span> <span class="n">np</span><span class="o">.</span><span class="n">all</span><span class="p">(</span><span class="n">h</span> <span class="o">==</span> <span class="n">mean</span><span class="p">[</span><span class="n">i</span><span class="p">]):</span>
|
||||
</span><span id="L-520"><a href="#L-520"><span class="linenos">520</span></a> <span class="k">continue</span>
|
||||
</span><span id="L-521"><a href="#L-521"><span class="linenos">521</span></a> <span class="n">deltas</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])</span>
|
||||
</span><span id="L-522"><a href="#L-522"><span class="linenos">522</span></a> <span class="n">obs_names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">name</span><span class="p">)</span>
|
||||
</span><span id="L-523"><a href="#L-523"><span class="linenos">523</span></a> <span class="n">idl</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">idld</span><span class="p">[</span><span class="n">name</span><span class="p">])</span>
|
||||
</span><span id="L-524"><a href="#L-524"><span class="linenos">524</span></a> <span class="n">res</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">deltas</span><span class="p">,</span> <span class="n">obs_names</span><span class="p">,</span> <span class="n">idl</span><span class="o">=</span><span class="n">idl</span><span class="p">))</span>
|
||||
</span><span id="L-525"><a href="#L-525"><span class="linenos">525</span></a> <span class="n">res</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">_value</span> <span class="o">=</span> <span class="n">mean</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
|
||||
</span><span id="L-526"><a href="#L-526"><span class="linenos">526</span></a> <span class="n">_check</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">e_names</span><span class="p">)</span> <span class="o">==</span> <span class="n">ne</span><span class="p">)</span>
|
||||
</span><span id="L-527"><a href="#L-527"><span class="linenos">527</span></a>
|
||||
</span><span id="L-528"><a href="#L-528"><span class="linenos">528</span></a> <span class="n">cnames</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="n">covd</span><span class="o">.</span><span class="n">keys</span><span class="p">())</span>
|
||||
</span><span id="L-529"><a href="#L-529"><span class="linenos">529</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">res</span><span class="p">)):</span>
|
||||
</span><span id="L-530"><a href="#L-530"><span class="linenos">530</span></a> <span class="n">new_covobs</span> <span class="o">=</span> <span class="p">{</span><span class="n">name</span><span class="p">:</span> <span class="n">Covobs</span><span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="n">covd</span><span class="p">[</span><span class="n">name</span><span class="p">],</span> <span class="n">name</span><span class="p">,</span> <span class="n">grad</span><span class="o">=</span><span class="n">gradd</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])</span> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">cnames</span><span class="p">}</span>
|
||||
</span><span id="L-531"><a href="#L-531"><span class="linenos">531</span></a> <span class="k">if</span> <span class="n">noempty</span><span class="p">:</span>
|
||||
</span><span id="L-532"><a href="#L-532"><span class="linenos">532</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">cnames</span><span class="p">:</span>
|
||||
</span><span id="L-533"><a href="#L-533"><span class="linenos">533</span></a> <span class="k">if</span> <span class="n">np</span><span class="o">.</span><span class="n">all</span><span class="p">(</span><span class="n">new_covobs</span><span class="p">[</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">grad</span> <span class="o">==</span> <span class="mi">0</span><span class="p">):</span>
|
||||
</span><span id="L-534"><a href="#L-534"><span class="linenos">534</span></a> <span class="k">del</span> <span class="n">new_covobs</span><span class="p">[</span><span class="n">name</span><span class="p">]</span>
|
||||
</span><span id="L-535"><a href="#L-535"><span class="linenos">535</span></a> <span class="n">cnames_loc</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="n">new_covobs</span><span class="o">.</span><span class="n">keys</span><span class="p">())</span>
|
||||
</span><span id="L-536"><a href="#L-536"><span class="linenos">536</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="L-537"><a href="#L-537"><span class="linenos">537</span></a> <span class="n">cnames_loc</span> <span class="o">=</span> <span class="n">cnames</span>
|
||||
</span><span id="L-538"><a href="#L-538"><span class="linenos">538</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">cnames_loc</span><span class="p">:</span>
|
||||
</span><span id="L-539"><a href="#L-539"><span class="linenos">539</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">name</span><span class="p">)</span>
|
||||
</span><span id="L-540"><a href="#L-540"><span class="linenos">540</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="mi">1</span>
|
||||
</span><span id="L-541"><a href="#L-541"><span class="linenos">541</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">idl</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="L-542"><a href="#L-542"><span class="linenos">542</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">_covobs</span> <span class="o">=</span> <span class="n">new_covobs</span>
|
||||
</span><span id="L-543"><a href="#L-543"><span class="linenos">543</span></a>
|
||||
</span><span id="L-544"><a href="#L-544"><span class="linenos">544</span></a> <span class="k">if</span> <span class="n">symbol</span><span class="p">:</span>
|
||||
</span><span id="L-545"><a href="#L-545"><span class="linenos">545</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">res</span><span class="p">)):</span>
|
||||
</span><span id="L-546"><a href="#L-546"><span class="linenos">546</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">=</span> <span class="n">symbol</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
|
||||
</span><span id="L-547"><a href="#L-547"><span class="linenos">547</span></a> <span class="k">if</span> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'None'</span><span class="p">:</span>
|
||||
</span><span id="L-548"><a href="#L-548"><span class="linenos">548</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">=</span> <span class="kc">None</span>
|
||||
</span><span id="L-549"><a href="#L-549"><span class="linenos">549</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">noempty</span><span class="p">:</span>
|
||||
</span><span id="L-550"><a href="#L-550"><span class="linenos">550</span></a> <span class="n">_check</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">res</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">covobs</span><span class="o">.</span><span class="n">keys</span><span class="p">())</span> <span class="o">==</span> <span class="n">nc</span><span class="p">)</span>
|
||||
</span><span id="L-551"><a href="#L-551"><span class="linenos">551</span></a> <span class="k">if</span> <span class="n">full_output</span><span class="p">:</span>
|
||||
</span><span id="L-552"><a href="#L-552"><span class="linenos">552</span></a> <span class="n">retd</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-553"><a href="#L-553"><span class="linenos">553</span></a> <span class="n">tool</span> <span class="o">=</span> <span class="n">file_origin</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s1">'tool'</span><span class="p">,</span> <span class="kc">None</span><span class="p">)</span>
|
||||
</span><span id="L-554"><a href="#L-554"><span class="linenos">554</span></a> <span class="k">if</span> <span class="n">tool</span><span class="p">:</span>
|
||||
</span><span id="L-555"><a href="#L-555"><span class="linenos">555</span></a> <span class="n">program</span> <span class="o">=</span> <span class="n">tool</span><span class="p">[</span><span class="s1">'name'</span><span class="p">]</span> <span class="o">+</span> <span class="s1">' '</span> <span class="o">+</span> <span class="n">tool</span><span class="p">[</span><span class="s1">'version'</span><span class="p">]</span>
|
||||
</span><span id="L-556"><a href="#L-556"><span class="linenos">556</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="L-557"><a href="#L-557"><span class="linenos">557</span></a> <span class="n">program</span> <span class="o">=</span> <span class="s1">''</span>
|
||||
</span><span id="L-558"><a href="#L-558"><span class="linenos">558</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'program'</span><span class="p">]</span> <span class="o">=</span> <span class="n">program</span>
|
||||
</span><span id="L-559"><a href="#L-559"><span class="linenos">559</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'version'</span><span class="p">]</span> <span class="o">=</span> <span class="n">version</span>
|
||||
</span><span id="L-560"><a href="#L-560"><span class="linenos">560</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'who'</span><span class="p">]</span> <span class="o">=</span> <span class="n">file_origin</span><span class="p">[</span><span class="s1">'who'</span><span class="p">]</span>
|
||||
</span><span id="L-561"><a href="#L-561"><span class="linenos">561</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'date'</span><span class="p">]</span> <span class="o">=</span> <span class="n">file_origin</span><span class="p">[</span><span class="s1">'date'</span><span class="p">]</span>
|
||||
</span><span id="L-562"><a href="#L-562"><span class="linenos">562</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'host'</span><span class="p">]</span> <span class="o">=</span> <span class="n">file_origin</span><span class="p">[</span><span class="s1">'host'</span><span class="p">]</span>
|
||||
</span><span id="L-563"><a href="#L-563"><span class="linenos">563</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'description'</span><span class="p">]</span> <span class="o">=</span> <span class="n">descriptiond</span>
|
||||
</span><span id="L-564"><a href="#L-564"><span class="linenos">564</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'enstags'</span><span class="p">]</span> <span class="o">=</span> <span class="n">enstags</span>
|
||||
</span><span id="L-565"><a href="#L-565"><span class="linenos">565</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'obsdata'</span><span class="p">]</span> <span class="o">=</span> <span class="n">res</span>
|
||||
</span><span id="L-566"><a href="#L-566"><span class="linenos">566</span></a> <span class="k">return</span> <span class="n">retd</span>
|
||||
</span><span id="L-567"><a href="#L-567"><span class="linenos">567</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="L-568"><a href="#L-568"><span class="linenos">568</span></a> <span class="k">return</span> <span class="n">res</span>
|
||||
</span><span id="L-569"><a href="#L-569"><span class="linenos">569</span></a>
|
||||
</span><span id="L-570"><a href="#L-570"><span class="linenos">570</span></a>
|
||||
</span><span id="L-571"><a href="#L-571"><span class="linenos">571</span></a><span class="k">def</span> <span class="nf">read_dobs</span><span class="p">(</span><span class="n">fname</span><span class="p">,</span> <span class="n">noempty</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">full_output</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">gz</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">separator_insertion</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>
|
||||
</span><span id="L-572"><a href="#L-572"><span class="linenos">572</span></a><span class="w"> </span><span class="sd">"""Import a list of Obs from an xml.gz file in the Zeuthen dobs format.</span>
|
||||
</span><span id="L-521"><a href="#L-521"><span class="linenos">521</span></a> <span class="n">repdeltas</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="L-522"><a href="#L-522"><span class="linenos">522</span></a> <span class="n">repidl</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="L-523"><a href="#L-523"><span class="linenos">523</span></a> <span class="k">for</span> <span class="n">j</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])):</span>
|
||||
</span><span id="L-524"><a href="#L-524"><span class="linenos">524</span></a> <span class="k">if</span> <span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">][</span><span class="n">j</span><span class="p">]</span> <span class="o">!=</span> <span class="mf">0.</span><span class="p">:</span>
|
||||
</span><span id="L-525"><a href="#L-525"><span class="linenos">525</span></a> <span class="n">repdeltas</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">][</span><span class="n">j</span><span class="p">])</span>
|
||||
</span><span id="L-526"><a href="#L-526"><span class="linenos">526</span></a> <span class="n">repidl</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">idld</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">j</span><span class="p">])</span>
|
||||
</span><span id="L-527"><a href="#L-527"><span class="linenos">527</span></a> <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">repdeltas</span><span class="p">)</span> <span class="o">></span> <span class="mi">0</span><span class="p">:</span>
|
||||
</span><span id="L-528"><a href="#L-528"><span class="linenos">528</span></a> <span class="n">obs_names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">name</span><span class="p">)</span>
|
||||
</span><span id="L-529"><a href="#L-529"><span class="linenos">529</span></a> <span class="n">deltas</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">repdeltas</span><span class="p">)</span>
|
||||
</span><span id="L-530"><a href="#L-530"><span class="linenos">530</span></a> <span class="n">idl</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">repidl</span><span class="p">)</span>
|
||||
</span><span id="L-531"><a href="#L-531"><span class="linenos">531</span></a>
|
||||
</span><span id="L-532"><a href="#L-532"><span class="linenos">532</span></a> <span class="n">res</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">deltas</span><span class="p">,</span> <span class="n">obs_names</span><span class="p">,</span> <span class="n">idl</span><span class="o">=</span><span class="n">idl</span><span class="p">))</span>
|
||||
</span><span id="L-533"><a href="#L-533"><span class="linenos">533</span></a> <span class="n">res</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">_value</span> <span class="o">=</span> <span class="n">mean</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
|
||||
</span><span id="L-534"><a href="#L-534"><span class="linenos">534</span></a> <span class="n">_check</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">e_names</span><span class="p">)</span> <span class="o">==</span> <span class="n">ne</span><span class="p">)</span>
|
||||
</span><span id="L-535"><a href="#L-535"><span class="linenos">535</span></a>
|
||||
</span><span id="L-536"><a href="#L-536"><span class="linenos">536</span></a> <span class="n">cnames</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="n">covd</span><span class="o">.</span><span class="n">keys</span><span class="p">())</span>
|
||||
</span><span id="L-537"><a href="#L-537"><span class="linenos">537</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">res</span><span class="p">)):</span>
|
||||
</span><span id="L-538"><a href="#L-538"><span class="linenos">538</span></a> <span class="n">new_covobs</span> <span class="o">=</span> <span class="p">{</span><span class="n">name</span><span class="p">:</span> <span class="n">Covobs</span><span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="n">covd</span><span class="p">[</span><span class="n">name</span><span class="p">],</span> <span class="n">name</span><span class="p">,</span> <span class="n">grad</span><span class="o">=</span><span class="n">gradd</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])</span> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">cnames</span><span class="p">}</span>
|
||||
</span><span id="L-539"><a href="#L-539"><span class="linenos">539</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">cnames</span><span class="p">:</span>
|
||||
</span><span id="L-540"><a href="#L-540"><span class="linenos">540</span></a> <span class="k">if</span> <span class="n">np</span><span class="o">.</span><span class="n">all</span><span class="p">(</span><span class="n">new_covobs</span><span class="p">[</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">grad</span> <span class="o">==</span> <span class="mi">0</span><span class="p">):</span>
|
||||
</span><span id="L-541"><a href="#L-541"><span class="linenos">541</span></a> <span class="k">del</span> <span class="n">new_covobs</span><span class="p">[</span><span class="n">name</span><span class="p">]</span>
|
||||
</span><span id="L-542"><a href="#L-542"><span class="linenos">542</span></a> <span class="n">cnames_loc</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="n">new_covobs</span><span class="o">.</span><span class="n">keys</span><span class="p">())</span>
|
||||
</span><span id="L-543"><a href="#L-543"><span class="linenos">543</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">cnames_loc</span><span class="p">:</span>
|
||||
</span><span id="L-544"><a href="#L-544"><span class="linenos">544</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">name</span><span class="p">)</span>
|
||||
</span><span id="L-545"><a href="#L-545"><span class="linenos">545</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="mi">1</span>
|
||||
</span><span id="L-546"><a href="#L-546"><span class="linenos">546</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">idl</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="L-547"><a href="#L-547"><span class="linenos">547</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">_covobs</span> <span class="o">=</span> <span class="n">new_covobs</span>
|
||||
</span><span id="L-548"><a href="#L-548"><span class="linenos">548</span></a>
|
||||
</span><span id="L-549"><a href="#L-549"><span class="linenos">549</span></a> <span class="k">if</span> <span class="n">symbol</span><span class="p">:</span>
|
||||
</span><span id="L-550"><a href="#L-550"><span class="linenos">550</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">res</span><span class="p">)):</span>
|
||||
</span><span id="L-551"><a href="#L-551"><span class="linenos">551</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">=</span> <span class="n">symbol</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
|
||||
</span><span id="L-552"><a href="#L-552"><span class="linenos">552</span></a> <span class="k">if</span> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'None'</span><span class="p">:</span>
|
||||
</span><span id="L-553"><a href="#L-553"><span class="linenos">553</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">=</span> <span class="kc">None</span>
|
||||
</span><span id="L-554"><a href="#L-554"><span class="linenos">554</span></a> <span class="k">if</span> <span class="n">full_output</span><span class="p">:</span>
|
||||
</span><span id="L-555"><a href="#L-555"><span class="linenos">555</span></a> <span class="n">retd</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="L-556"><a href="#L-556"><span class="linenos">556</span></a> <span class="n">tool</span> <span class="o">=</span> <span class="n">file_origin</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s1">'tool'</span><span class="p">,</span> <span class="kc">None</span><span class="p">)</span>
|
||||
</span><span id="L-557"><a href="#L-557"><span class="linenos">557</span></a> <span class="k">if</span> <span class="n">tool</span><span class="p">:</span>
|
||||
</span><span id="L-558"><a href="#L-558"><span class="linenos">558</span></a> <span class="n">program</span> <span class="o">=</span> <span class="n">tool</span><span class="p">[</span><span class="s1">'name'</span><span class="p">]</span> <span class="o">+</span> <span class="s1">' '</span> <span class="o">+</span> <span class="n">tool</span><span class="p">[</span><span class="s1">'version'</span><span class="p">]</span>
|
||||
</span><span id="L-559"><a href="#L-559"><span class="linenos">559</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="L-560"><a href="#L-560"><span class="linenos">560</span></a> <span class="n">program</span> <span class="o">=</span> <span class="s1">''</span>
|
||||
</span><span id="L-561"><a href="#L-561"><span class="linenos">561</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'program'</span><span class="p">]</span> <span class="o">=</span> <span class="n">program</span>
|
||||
</span><span id="L-562"><a href="#L-562"><span class="linenos">562</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'version'</span><span class="p">]</span> <span class="o">=</span> <span class="n">version</span>
|
||||
</span><span id="L-563"><a href="#L-563"><span class="linenos">563</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'who'</span><span class="p">]</span> <span class="o">=</span> <span class="n">file_origin</span><span class="p">[</span><span class="s1">'who'</span><span class="p">]</span>
|
||||
</span><span id="L-564"><a href="#L-564"><span class="linenos">564</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'date'</span><span class="p">]</span> <span class="o">=</span> <span class="n">file_origin</span><span class="p">[</span><span class="s1">'date'</span><span class="p">]</span>
|
||||
</span><span id="L-565"><a href="#L-565"><span class="linenos">565</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'host'</span><span class="p">]</span> <span class="o">=</span> <span class="n">file_origin</span><span class="p">[</span><span class="s1">'host'</span><span class="p">]</span>
|
||||
</span><span id="L-566"><a href="#L-566"><span class="linenos">566</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'description'</span><span class="p">]</span> <span class="o">=</span> <span class="n">descriptiond</span>
|
||||
</span><span id="L-567"><a href="#L-567"><span class="linenos">567</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'enstags'</span><span class="p">]</span> <span class="o">=</span> <span class="n">enstags</span>
|
||||
</span><span id="L-568"><a href="#L-568"><span class="linenos">568</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'obsdata'</span><span class="p">]</span> <span class="o">=</span> <span class="n">res</span>
|
||||
</span><span id="L-569"><a href="#L-569"><span class="linenos">569</span></a> <span class="k">return</span> <span class="n">retd</span>
|
||||
</span><span id="L-570"><a href="#L-570"><span class="linenos">570</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="L-571"><a href="#L-571"><span class="linenos">571</span></a> <span class="k">return</span> <span class="n">res</span>
|
||||
</span><span id="L-572"><a href="#L-572"><span class="linenos">572</span></a>
|
||||
</span><span id="L-573"><a href="#L-573"><span class="linenos">573</span></a>
|
||||
</span><span id="L-574"><a href="#L-574"><span class="linenos">574</span></a><span class="sd"> Tags are not written or recovered automatically.</span>
|
||||
</span><span id="L-575"><a href="#L-575"><span class="linenos">575</span></a>
|
||||
</span><span id="L-576"><a href="#L-576"><span class="linenos">576</span></a><span class="sd"> Parameters</span>
|
||||
</span><span id="L-577"><a href="#L-577"><span class="linenos">577</span></a><span class="sd"> ----------</span>
|
||||
</span><span id="L-578"><a href="#L-578"><span class="linenos">578</span></a><span class="sd"> fname : str</span>
|
||||
</span><span id="L-579"><a href="#L-579"><span class="linenos">579</span></a><span class="sd"> Filename of the input file.</span>
|
||||
</span><span id="L-580"><a href="#L-580"><span class="linenos">580</span></a><span class="sd"> noemtpy : bool</span>
|
||||
</span><span id="L-581"><a href="#L-581"><span class="linenos">581</span></a><span class="sd"> If True, ensembles with no contribution to the Obs are not included.</span>
|
||||
</span><span id="L-582"><a href="#L-582"><span class="linenos">582</span></a><span class="sd"> If False, ensembles are included as written in the file.</span>
|
||||
</span><span id="L-574"><a href="#L-574"><span class="linenos">574</span></a><span class="k">def</span> <span class="nf">read_dobs</span><span class="p">(</span><span class="n">fname</span><span class="p">,</span> <span class="n">full_output</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">gz</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">separator_insertion</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>
|
||||
</span><span id="L-575"><a href="#L-575"><span class="linenos">575</span></a><span class="w"> </span><span class="sd">"""Import a list of Obs from an xml.gz file in the Zeuthen dobs format.</span>
|
||||
</span><span id="L-576"><a href="#L-576"><span class="linenos">576</span></a>
|
||||
</span><span id="L-577"><a href="#L-577"><span class="linenos">577</span></a><span class="sd"> Tags are not written or recovered automatically.</span>
|
||||
</span><span id="L-578"><a href="#L-578"><span class="linenos">578</span></a>
|
||||
</span><span id="L-579"><a href="#L-579"><span class="linenos">579</span></a><span class="sd"> Parameters</span>
|
||||
</span><span id="L-580"><a href="#L-580"><span class="linenos">580</span></a><span class="sd"> ----------</span>
|
||||
</span><span id="L-581"><a href="#L-581"><span class="linenos">581</span></a><span class="sd"> fname : str</span>
|
||||
</span><span id="L-582"><a href="#L-582"><span class="linenos">582</span></a><span class="sd"> Filename of the input file.</span>
|
||||
</span><span id="L-583"><a href="#L-583"><span class="linenos">583</span></a><span class="sd"> full_output : bool</span>
|
||||
</span><span id="L-584"><a href="#L-584"><span class="linenos">584</span></a><span class="sd"> If True, a dict containing auxiliary information and the data is returned.</span>
|
||||
</span><span id="L-585"><a href="#L-585"><span class="linenos">585</span></a><span class="sd"> If False, only the data is returned as list.</span>
|
||||
|
|
@ -711,7 +711,7 @@
|
|||
</span><span id="L-615"><a href="#L-615"><span class="linenos">615</span></a> <span class="k">with</span> <span class="nb">open</span><span class="p">(</span><span class="n">fname</span><span class="p">,</span> <span class="s1">'r'</span><span class="p">)</span> <span class="k">as</span> <span class="n">fin</span><span class="p">:</span>
|
||||
</span><span id="L-616"><a href="#L-616"><span class="linenos">616</span></a> <span class="n">content</span> <span class="o">=</span> <span class="n">fin</span><span class="o">.</span><span class="n">read</span><span class="p">()</span>
|
||||
</span><span id="L-617"><a href="#L-617"><span class="linenos">617</span></a>
|
||||
</span><span id="L-618"><a href="#L-618"><span class="linenos">618</span></a> <span class="k">return</span> <span class="n">import_dobs_string</span><span class="p">(</span><span class="n">content</span><span class="p">,</span> <span class="n">noempty</span><span class="p">,</span> <span class="n">full_output</span><span class="p">,</span> <span class="n">separator_insertion</span><span class="o">=</span><span class="n">separator_insertion</span><span class="p">)</span>
|
||||
</span><span id="L-618"><a href="#L-618"><span class="linenos">618</span></a> <span class="k">return</span> <span class="n">import_dobs_string</span><span class="p">(</span><span class="n">content</span><span class="p">,</span> <span class="n">full_output</span><span class="p">,</span> <span class="n">separator_insertion</span><span class="o">=</span><span class="n">separator_insertion</span><span class="p">)</span>
|
||||
</span><span id="L-619"><a href="#L-619"><span class="linenos">619</span></a>
|
||||
</span><span id="L-620"><a href="#L-620"><span class="linenos">620</span></a>
|
||||
</span><span id="L-621"><a href="#L-621"><span class="linenos">621</span></a><span class="k">def</span> <span class="nf">_dobsdict_to_xmlstring</span><span class="p">(</span><span class="n">d</span><span class="p">):</span>
|
||||
|
|
@ -878,7 +878,7 @@
|
|||
</span><span id="L-782"><a href="#L-782"><span class="linenos">782</span></a> <span class="n">o</span> <span class="o">=</span> <span class="n">obsl</span><span class="p">[</span><span class="n">oi</span><span class="p">]</span>
|
||||
</span><span id="L-783"><a href="#L-783"><span class="linenos">783</span></a> <span class="k">if</span> <span class="n">repname</span> <span class="ow">in</span> <span class="n">o</span><span class="o">.</span><span class="n">idl</span><span class="p">:</span>
|
||||
</span><span id="L-784"><a href="#L-784"><span class="linenos">784</span></a> <span class="k">if</span> <span class="n">counters</span><span class="p">[</span><span class="n">oi</span><span class="p">]</span> <span class="o"><</span> <span class="mi">0</span><span class="p">:</span>
|
||||
</span><span id="L-785"><a href="#L-785"><span class="linenos">785</span></a> <span class="n">num</span> <span class="o">=</span> <span class="n">offsets</span><span class="p">[</span><span class="n">oi</span><span class="p">]</span>
|
||||
</span><span id="L-785"><a href="#L-785"><span class="linenos">785</span></a> <span class="n">num</span> <span class="o">=</span> <span class="mi">0</span>
|
||||
</span><span id="L-786"><a href="#L-786"><span class="linenos">786</span></a> <span class="k">if</span> <span class="n">num</span> <span class="o">==</span> <span class="mi">0</span><span class="p">:</span>
|
||||
</span><span id="L-787"><a href="#L-787"><span class="linenos">787</span></a> <span class="n">data</span> <span class="o">+=</span> <span class="s1">'0 '</span>
|
||||
</span><span id="L-788"><a href="#L-788"><span class="linenos">788</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
|
|
@ -894,7 +894,7 @@
|
|||
</span><span id="L-798"><a href="#L-798"><span class="linenos">798</span></a> <span class="k">if</span> <span class="n">counters</span><span class="p">[</span><span class="n">oi</span><span class="p">]</span> <span class="o">>=</span> <span class="nb">len</span><span class="p">(</span><span class="n">o</span><span class="o">.</span><span class="n">idl</span><span class="p">[</span><span class="n">repname</span><span class="p">]):</span>
|
||||
</span><span id="L-799"><a href="#L-799"><span class="linenos">799</span></a> <span class="n">counters</span><span class="p">[</span><span class="n">oi</span><span class="p">]</span> <span class="o">=</span> <span class="o">-</span><span class="mi">1</span>
|
||||
</span><span id="L-800"><a href="#L-800"><span class="linenos">800</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="L-801"><a href="#L-801"><span class="linenos">801</span></a> <span class="n">num</span> <span class="o">=</span> <span class="n">offsets</span><span class="p">[</span><span class="n">oi</span><span class="p">]</span>
|
||||
</span><span id="L-801"><a href="#L-801"><span class="linenos">801</span></a> <span class="n">num</span> <span class="o">=</span> <span class="mi">0</span>
|
||||
</span><span id="L-802"><a href="#L-802"><span class="linenos">802</span></a> <span class="k">if</span> <span class="n">num</span> <span class="o">==</span> <span class="mi">0</span><span class="p">:</span>
|
||||
</span><span id="L-803"><a href="#L-803"><span class="linenos">803</span></a> <span class="n">data</span> <span class="o">+=</span> <span class="s1">'0 '</span>
|
||||
</span><span id="L-804"><a href="#L-804"><span class="linenos">804</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
|
|
@ -1397,13 +1397,13 @@ Imported data and meta-data</li>
|
|||
<div class="attr function">
|
||||
|
||||
<span class="def">def</span>
|
||||
<span class="name">import_dobs_string</span><span class="signature pdoc-code condensed">(<span class="param"><span class="n">content</span>, </span><span class="param"><span class="n">noempty</span><span class="o">=</span><span class="kc">False</span>, </span><span class="param"><span class="n">full_output</span><span class="o">=</span><span class="kc">False</span>, </span><span class="param"><span class="n">separator_insertion</span><span class="o">=</span><span class="kc">True</span></span><span class="return-annotation">):</span></span>
|
||||
<span class="name">import_dobs_string</span><span class="signature pdoc-code condensed">(<span class="param"><span class="n">content</span>, </span><span class="param"><span class="n">full_output</span><span class="o">=</span><span class="kc">False</span>, </span><span class="param"><span class="n">separator_insertion</span><span class="o">=</span><span class="kc">True</span></span><span class="return-annotation">):</span></span>
|
||||
|
||||
<label class="view-source-button" for="import_dobs_string-view-source"><span>View Source</span></label>
|
||||
|
||||
</div>
|
||||
<a class="headerlink" href="#import_dobs_string"></a>
|
||||
<div class="pdoc-code codehilite"><pre><span></span><span id="import_dobs_string-401"><a href="#import_dobs_string-401"><span class="linenos">401</span></a><span class="k">def</span> <span class="nf">import_dobs_string</span><span class="p">(</span><span class="n">content</span><span class="p">,</span> <span class="n">noempty</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">full_output</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">separator_insertion</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>
|
||||
<div class="pdoc-code codehilite"><pre><span></span><span id="import_dobs_string-401"><a href="#import_dobs_string-401"><span class="linenos">401</span></a><span class="k">def</span> <span class="nf">import_dobs_string</span><span class="p">(</span><span class="n">content</span><span class="p">,</span> <span class="n">full_output</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">separator_insertion</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>
|
||||
</span><span id="import_dobs_string-402"><a href="#import_dobs_string-402"><span class="linenos">402</span></a><span class="w"> </span><span class="sd">"""Import a list of Obs from a string in the Zeuthen dobs format.</span>
|
||||
</span><span id="import_dobs_string-403"><a href="#import_dobs_string-403"><span class="linenos">403</span></a>
|
||||
</span><span id="import_dobs_string-404"><a href="#import_dobs_string-404"><span class="linenos">404</span></a><span class="sd"> Tags are not written or recovered automatically.</span>
|
||||
|
|
@ -1412,108 +1412,108 @@ Imported data and meta-data</li>
|
|||
</span><span id="import_dobs_string-407"><a href="#import_dobs_string-407"><span class="linenos">407</span></a><span class="sd"> ----------</span>
|
||||
</span><span id="import_dobs_string-408"><a href="#import_dobs_string-408"><span class="linenos">408</span></a><span class="sd"> content : str</span>
|
||||
</span><span id="import_dobs_string-409"><a href="#import_dobs_string-409"><span class="linenos">409</span></a><span class="sd"> XML string containing the data</span>
|
||||
</span><span id="import_dobs_string-410"><a href="#import_dobs_string-410"><span class="linenos">410</span></a><span class="sd"> noemtpy : bool</span>
|
||||
</span><span id="import_dobs_string-411"><a href="#import_dobs_string-411"><span class="linenos">411</span></a><span class="sd"> If True, ensembles with no contribution to the Obs are not included.</span>
|
||||
</span><span id="import_dobs_string-412"><a href="#import_dobs_string-412"><span class="linenos">412</span></a><span class="sd"> If False, ensembles are included as written in the file, possibly with vanishing entries.</span>
|
||||
</span><span id="import_dobs_string-413"><a href="#import_dobs_string-413"><span class="linenos">413</span></a><span class="sd"> full_output : bool</span>
|
||||
</span><span id="import_dobs_string-414"><a href="#import_dobs_string-414"><span class="linenos">414</span></a><span class="sd"> If True, a dict containing auxiliary information and the data is returned.</span>
|
||||
</span><span id="import_dobs_string-415"><a href="#import_dobs_string-415"><span class="linenos">415</span></a><span class="sd"> If False, only the data is returned as list.</span>
|
||||
</span><span id="import_dobs_string-416"><a href="#import_dobs_string-416"><span class="linenos">416</span></a><span class="sd"> separatior_insertion: str, int or bool</span>
|
||||
</span><span id="import_dobs_string-417"><a href="#import_dobs_string-417"><span class="linenos">417</span></a><span class="sd"> str: replace all occurences of "separator_insertion" within the replica names</span>
|
||||
</span><span id="import_dobs_string-418"><a href="#import_dobs_string-418"><span class="linenos">418</span></a><span class="sd"> by "|%s" % (separator_insertion) when constructing the names of the replica.</span>
|
||||
</span><span id="import_dobs_string-419"><a href="#import_dobs_string-419"><span class="linenos">419</span></a><span class="sd"> int: Insert the separator "|" at the position given by separator_insertion.</span>
|
||||
</span><span id="import_dobs_string-420"><a href="#import_dobs_string-420"><span class="linenos">420</span></a><span class="sd"> True (default): separator "|" is inserted after len(ensname), assuming that the</span>
|
||||
</span><span id="import_dobs_string-421"><a href="#import_dobs_string-421"><span class="linenos">421</span></a><span class="sd"> ensemble name is a prefix to the replica name.</span>
|
||||
</span><span id="import_dobs_string-422"><a href="#import_dobs_string-422"><span class="linenos">422</span></a><span class="sd"> None or False: No separator is inserted.</span>
|
||||
</span><span id="import_dobs_string-423"><a href="#import_dobs_string-423"><span class="linenos">423</span></a>
|
||||
</span><span id="import_dobs_string-424"><a href="#import_dobs_string-424"><span class="linenos">424</span></a><span class="sd"> Returns</span>
|
||||
</span><span id="import_dobs_string-425"><a href="#import_dobs_string-425"><span class="linenos">425</span></a><span class="sd"> -------</span>
|
||||
</span><span id="import_dobs_string-426"><a href="#import_dobs_string-426"><span class="linenos">426</span></a><span class="sd"> res : list[Obs]</span>
|
||||
</span><span id="import_dobs_string-427"><a href="#import_dobs_string-427"><span class="linenos">427</span></a><span class="sd"> Imported data</span>
|
||||
</span><span id="import_dobs_string-428"><a href="#import_dobs_string-428"><span class="linenos">428</span></a><span class="sd"> or</span>
|
||||
</span><span id="import_dobs_string-429"><a href="#import_dobs_string-429"><span class="linenos">429</span></a><span class="sd"> res : dict</span>
|
||||
</span><span id="import_dobs_string-430"><a href="#import_dobs_string-430"><span class="linenos">430</span></a><span class="sd"> Imported data and meta-data</span>
|
||||
</span><span id="import_dobs_string-431"><a href="#import_dobs_string-431"><span class="linenos">431</span></a><span class="sd"> """</span>
|
||||
</span><span id="import_dobs_string-432"><a href="#import_dobs_string-432"><span class="linenos">432</span></a>
|
||||
</span><span id="import_dobs_string-433"><a href="#import_dobs_string-433"><span class="linenos">433</span></a> <span class="n">root</span> <span class="o">=</span> <span class="n">et</span><span class="o">.</span><span class="n">fromstring</span><span class="p">(</span><span class="n">content</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-434"><a href="#import_dobs_string-434"><span class="linenos">434</span></a>
|
||||
</span><span id="import_dobs_string-435"><a href="#import_dobs_string-435"><span class="linenos">435</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'OBSERVABLES'</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-436"><a href="#import_dobs_string-436"><span class="linenos">436</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'SCHEMA'</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-437"><a href="#import_dobs_string-437"><span class="linenos">437</span></a> <span class="n">version</span> <span class="o">=</span> <span class="n">root</span><span class="p">[</span><span class="mi">0</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="import_dobs_string-410"><a href="#import_dobs_string-410"><span class="linenos">410</span></a><span class="sd"> full_output : bool</span>
|
||||
</span><span id="import_dobs_string-411"><a href="#import_dobs_string-411"><span class="linenos">411</span></a><span class="sd"> If True, a dict containing auxiliary information and the data is returned.</span>
|
||||
</span><span id="import_dobs_string-412"><a href="#import_dobs_string-412"><span class="linenos">412</span></a><span class="sd"> If False, only the data is returned as list.</span>
|
||||
</span><span id="import_dobs_string-413"><a href="#import_dobs_string-413"><span class="linenos">413</span></a><span class="sd"> separatior_insertion: str, int or bool</span>
|
||||
</span><span id="import_dobs_string-414"><a href="#import_dobs_string-414"><span class="linenos">414</span></a><span class="sd"> str: replace all occurences of "separator_insertion" within the replica names</span>
|
||||
</span><span id="import_dobs_string-415"><a href="#import_dobs_string-415"><span class="linenos">415</span></a><span class="sd"> by "|%s" % (separator_insertion) when constructing the names of the replica.</span>
|
||||
</span><span id="import_dobs_string-416"><a href="#import_dobs_string-416"><span class="linenos">416</span></a><span class="sd"> int: Insert the separator "|" at the position given by separator_insertion.</span>
|
||||
</span><span id="import_dobs_string-417"><a href="#import_dobs_string-417"><span class="linenos">417</span></a><span class="sd"> True (default): separator "|" is inserted after len(ensname), assuming that the</span>
|
||||
</span><span id="import_dobs_string-418"><a href="#import_dobs_string-418"><span class="linenos">418</span></a><span class="sd"> ensemble name is a prefix to the replica name.</span>
|
||||
</span><span id="import_dobs_string-419"><a href="#import_dobs_string-419"><span class="linenos">419</span></a><span class="sd"> None or False: No separator is inserted.</span>
|
||||
</span><span id="import_dobs_string-420"><a href="#import_dobs_string-420"><span class="linenos">420</span></a>
|
||||
</span><span id="import_dobs_string-421"><a href="#import_dobs_string-421"><span class="linenos">421</span></a><span class="sd"> Returns</span>
|
||||
</span><span id="import_dobs_string-422"><a href="#import_dobs_string-422"><span class="linenos">422</span></a><span class="sd"> -------</span>
|
||||
</span><span id="import_dobs_string-423"><a href="#import_dobs_string-423"><span class="linenos">423</span></a><span class="sd"> res : list[Obs]</span>
|
||||
</span><span id="import_dobs_string-424"><a href="#import_dobs_string-424"><span class="linenos">424</span></a><span class="sd"> Imported data</span>
|
||||
</span><span id="import_dobs_string-425"><a href="#import_dobs_string-425"><span class="linenos">425</span></a><span class="sd"> or</span>
|
||||
</span><span id="import_dobs_string-426"><a href="#import_dobs_string-426"><span class="linenos">426</span></a><span class="sd"> res : dict</span>
|
||||
</span><span id="import_dobs_string-427"><a href="#import_dobs_string-427"><span class="linenos">427</span></a><span class="sd"> Imported data and meta-data</span>
|
||||
</span><span id="import_dobs_string-428"><a href="#import_dobs_string-428"><span class="linenos">428</span></a><span class="sd"> """</span>
|
||||
</span><span id="import_dobs_string-429"><a href="#import_dobs_string-429"><span class="linenos">429</span></a>
|
||||
</span><span id="import_dobs_string-430"><a href="#import_dobs_string-430"><span class="linenos">430</span></a> <span class="n">root</span> <span class="o">=</span> <span class="n">et</span><span class="o">.</span><span class="n">fromstring</span><span class="p">(</span><span class="n">content</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-431"><a href="#import_dobs_string-431"><span class="linenos">431</span></a>
|
||||
</span><span id="import_dobs_string-432"><a href="#import_dobs_string-432"><span class="linenos">432</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'OBSERVABLES'</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-433"><a href="#import_dobs_string-433"><span class="linenos">433</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'SCHEMA'</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-434"><a href="#import_dobs_string-434"><span class="linenos">434</span></a> <span class="n">version</span> <span class="o">=</span> <span class="n">root</span><span class="p">[</span><span class="mi">0</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="import_dobs_string-435"><a href="#import_dobs_string-435"><span class="linenos">435</span></a>
|
||||
</span><span id="import_dobs_string-436"><a href="#import_dobs_string-436"><span class="linenos">436</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'origin'</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-437"><a href="#import_dobs_string-437"><span class="linenos">437</span></a> <span class="n">file_origin</span> <span class="o">=</span> <span class="n">_etree_to_dict</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">1</span><span class="p">])[</span><span class="s1">'origin'</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-438"><a href="#import_dobs_string-438"><span class="linenos">438</span></a>
|
||||
</span><span id="import_dobs_string-439"><a href="#import_dobs_string-439"><span class="linenos">439</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'origin'</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-440"><a href="#import_dobs_string-440"><span class="linenos">440</span></a> <span class="n">file_origin</span> <span class="o">=</span> <span class="n">_etree_to_dict</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">1</span><span class="p">])[</span><span class="s1">'origin'</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-441"><a href="#import_dobs_string-441"><span class="linenos">441</span></a>
|
||||
</span><span id="import_dobs_string-442"><a href="#import_dobs_string-442"><span class="linenos">442</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'dobs'</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-443"><a href="#import_dobs_string-443"><span class="linenos">443</span></a>
|
||||
</span><span id="import_dobs_string-444"><a href="#import_dobs_string-444"><span class="linenos">444</span></a> <span class="n">dobs</span> <span class="o">=</span> <span class="n">root</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-445"><a href="#import_dobs_string-445"><span class="linenos">445</span></a>
|
||||
</span><span id="import_dobs_string-446"><a href="#import_dobs_string-446"><span class="linenos">446</span></a> <span class="n">descriptiond</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-447"><a href="#import_dobs_string-447"><span class="linenos">447</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">3</span><span class="p">):</span>
|
||||
</span><span id="import_dobs_string-448"><a href="#import_dobs_string-448"><span class="linenos">448</span></a> <span class="n">descriptiond</span><span class="p">[</span><span class="n">dobs</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span><span class="p">]</span> <span class="o">=</span> <span class="n">dobs</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="import_dobs_string-449"><a href="#import_dobs_string-449"><span class="linenos">449</span></a>
|
||||
</span><span id="import_dobs_string-450"><a href="#import_dobs_string-450"><span class="linenos">450</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'array'</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-451"><a href="#import_dobs_string-451"><span class="linenos">451</span></a>
|
||||
</span><span id="import_dobs_string-452"><a href="#import_dobs_string-452"><span class="linenos">452</span></a> <span class="n">symbol</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="import_dobs_string-453"><a href="#import_dobs_string-453"><span class="linenos">453</span></a> <span class="k">if</span> <span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'symbol'</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-454"><a href="#import_dobs_string-454"><span class="linenos">454</span></a> <span class="n">symbol</span> <span class="o">=</span> <span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="import_dobs_string-455"><a href="#import_dobs_string-455"><span class="linenos">455</span></a> <span class="n">descriptiond</span><span class="p">[</span><span class="s1">'symbol'</span><span class="p">]</span> <span class="o">=</span> <span class="n">symbol</span>
|
||||
</span><span id="import_dobs_string-456"><a href="#import_dobs_string-456"><span class="linenos">456</span></a> <span class="n">mean</span> <span class="o">=</span> <span class="n">_import_array</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">])[</span><span class="mi">0</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-457"><a href="#import_dobs_string-457"><span class="linenos">457</span></a>
|
||||
</span><span id="import_dobs_string-458"><a href="#import_dobs_string-458"><span class="linenos">458</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">4</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"ne"</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-459"><a href="#import_dobs_string-459"><span class="linenos">459</span></a> <span class="n">ne</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">4</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">())</span>
|
||||
</span><span id="import_dobs_string-460"><a href="#import_dobs_string-460"><span class="linenos">460</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">5</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"nc"</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-461"><a href="#import_dobs_string-461"><span class="linenos">461</span></a> <span class="n">nc</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">5</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">())</span>
|
||||
</span><span id="import_dobs_string-462"><a href="#import_dobs_string-462"><span class="linenos">462</span></a>
|
||||
</span><span id="import_dobs_string-463"><a href="#import_dobs_string-463"><span class="linenos">463</span></a> <span class="n">idld</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-464"><a href="#import_dobs_string-464"><span class="linenos">464</span></a> <span class="n">deltad</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-465"><a href="#import_dobs_string-465"><span class="linenos">465</span></a> <span class="n">covd</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-466"><a href="#import_dobs_string-466"><span class="linenos">466</span></a> <span class="n">gradd</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-467"><a href="#import_dobs_string-467"><span class="linenos">467</span></a> <span class="n">names</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="import_dobs_string-468"><a href="#import_dobs_string-468"><span class="linenos">468</span></a> <span class="n">e_names</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="import_dobs_string-469"><a href="#import_dobs_string-469"><span class="linenos">469</span></a> <span class="n">enstags</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-470"><a href="#import_dobs_string-470"><span class="linenos">470</span></a> <span class="k">for</span> <span class="n">k</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">6</span><span class="p">,</span> <span class="nb">len</span><span class="p">(</span><span class="nb">list</span><span class="p">(</span><span class="n">dobs</span><span class="p">))):</span>
|
||||
</span><span id="import_dobs_string-471"><a href="#import_dobs_string-471"><span class="linenos">471</span></a> <span class="k">if</span> <span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"edata"</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-472"><a href="#import_dobs_string-472"><span class="linenos">472</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"enstag"</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-473"><a href="#import_dobs_string-473"><span class="linenos">473</span></a> <span class="n">ename</span> <span class="o">=</span> <span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="import_dobs_string-474"><a href="#import_dobs_string-474"><span class="linenos">474</span></a> <span class="n">e_names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">ename</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-475"><a href="#import_dobs_string-475"><span class="linenos">475</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"nr"</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-476"><a href="#import_dobs_string-476"><span class="linenos">476</span></a> <span class="n">R</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">())</span>
|
||||
</span><span id="import_dobs_string-477"><a href="#import_dobs_string-477"><span class="linenos">477</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">2</span><span class="p">,</span> <span class="mi">2</span> <span class="o">+</span> <span class="n">R</span><span class="p">):</span>
|
||||
</span><span id="import_dobs_string-478"><a href="#import_dobs_string-478"><span class="linenos">478</span></a> <span class="n">deltas</span><span class="p">,</span> <span class="n">rname</span><span class="p">,</span> <span class="n">idx</span> <span class="o">=</span> <span class="n">_import_rdata</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="n">i</span><span class="p">])</span>
|
||||
</span><span id="import_dobs_string-479"><a href="#import_dobs_string-479"><span class="linenos">479</span></a> <span class="k">if</span> <span class="n">separator_insertion</span> <span class="ow">is</span> <span class="kc">None</span> <span class="ow">or</span> <span class="kc">False</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-480"><a href="#import_dobs_string-480"><span class="linenos">480</span></a> <span class="k">pass</span>
|
||||
</span><span id="import_dobs_string-481"><a href="#import_dobs_string-481"><span class="linenos">481</span></a> <span class="k">elif</span> <span class="n">separator_insertion</span> <span class="ow">is</span> <span class="kc">True</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-482"><a href="#import_dobs_string-482"><span class="linenos">482</span></a> <span class="k">if</span> <span class="n">rname</span><span class="o">.</span><span class="n">startswith</span><span class="p">(</span><span class="n">ename</span><span class="p">):</span>
|
||||
</span><span id="import_dobs_string-483"><a href="#import_dobs_string-483"><span class="linenos">483</span></a> <span class="n">rname</span> <span class="o">=</span> <span class="n">rname</span><span class="p">[:</span><span class="nb">len</span><span class="p">(</span><span class="n">ename</span><span class="p">)]</span> <span class="o">+</span> <span class="s1">'|'</span> <span class="o">+</span> <span class="n">rname</span><span class="p">[</span><span class="nb">len</span><span class="p">(</span><span class="n">ename</span><span class="p">):]</span>
|
||||
</span><span id="import_dobs_string-484"><a href="#import_dobs_string-484"><span class="linenos">484</span></a> <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">,</span> <span class="nb">int</span><span class="p">):</span>
|
||||
</span><span id="import_dobs_string-485"><a href="#import_dobs_string-485"><span class="linenos">485</span></a> <span class="n">rname</span> <span class="o">=</span> <span class="n">rname</span><span class="p">[:</span><span class="n">separator_insertion</span><span class="p">]</span> <span class="o">+</span> <span class="s1">'|'</span> <span class="o">+</span> <span class="n">rname</span><span class="p">[</span><span class="n">separator_insertion</span><span class="p">:]</span>
|
||||
</span><span id="import_dobs_string-486"><a href="#import_dobs_string-486"><span class="linenos">486</span></a> <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">,</span> <span class="nb">str</span><span class="p">):</span>
|
||||
</span><span id="import_dobs_string-487"><a href="#import_dobs_string-487"><span class="linenos">487</span></a> <span class="n">rname</span> <span class="o">=</span> <span class="n">rname</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">,</span> <span class="s2">"|</span><span class="si">%s</span><span class="s2">"</span> <span class="o">%</span> <span class="p">(</span><span class="n">separator_insertion</span><span class="p">))</span>
|
||||
</span><span id="import_dobs_string-439"><a href="#import_dobs_string-439"><span class="linenos">439</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">root</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'dobs'</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-440"><a href="#import_dobs_string-440"><span class="linenos">440</span></a>
|
||||
</span><span id="import_dobs_string-441"><a href="#import_dobs_string-441"><span class="linenos">441</span></a> <span class="n">dobs</span> <span class="o">=</span> <span class="n">root</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-442"><a href="#import_dobs_string-442"><span class="linenos">442</span></a>
|
||||
</span><span id="import_dobs_string-443"><a href="#import_dobs_string-443"><span class="linenos">443</span></a> <span class="n">descriptiond</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-444"><a href="#import_dobs_string-444"><span class="linenos">444</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">3</span><span class="p">):</span>
|
||||
</span><span id="import_dobs_string-445"><a href="#import_dobs_string-445"><span class="linenos">445</span></a> <span class="n">descriptiond</span><span class="p">[</span><span class="n">dobs</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span><span class="p">]</span> <span class="o">=</span> <span class="n">dobs</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="import_dobs_string-446"><a href="#import_dobs_string-446"><span class="linenos">446</span></a>
|
||||
</span><span id="import_dobs_string-447"><a href="#import_dobs_string-447"><span class="linenos">447</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'array'</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-448"><a href="#import_dobs_string-448"><span class="linenos">448</span></a>
|
||||
</span><span id="import_dobs_string-449"><a href="#import_dobs_string-449"><span class="linenos">449</span></a> <span class="n">symbol</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="import_dobs_string-450"><a href="#import_dobs_string-450"><span class="linenos">450</span></a> <span class="k">if</span> <span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'symbol'</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-451"><a href="#import_dobs_string-451"><span class="linenos">451</span></a> <span class="n">symbol</span> <span class="o">=</span> <span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="import_dobs_string-452"><a href="#import_dobs_string-452"><span class="linenos">452</span></a> <span class="n">descriptiond</span><span class="p">[</span><span class="s1">'symbol'</span><span class="p">]</span> <span class="o">=</span> <span class="n">symbol</span>
|
||||
</span><span id="import_dobs_string-453"><a href="#import_dobs_string-453"><span class="linenos">453</span></a> <span class="n">mean</span> <span class="o">=</span> <span class="n">_import_array</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">3</span><span class="p">])[</span><span class="mi">0</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-454"><a href="#import_dobs_string-454"><span class="linenos">454</span></a>
|
||||
</span><span id="import_dobs_string-455"><a href="#import_dobs_string-455"><span class="linenos">455</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">4</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"ne"</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-456"><a href="#import_dobs_string-456"><span class="linenos">456</span></a> <span class="n">ne</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">4</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">())</span>
|
||||
</span><span id="import_dobs_string-457"><a href="#import_dobs_string-457"><span class="linenos">457</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="mi">5</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"nc"</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-458"><a href="#import_dobs_string-458"><span class="linenos">458</span></a>
|
||||
</span><span id="import_dobs_string-459"><a href="#import_dobs_string-459"><span class="linenos">459</span></a> <span class="n">idld</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-460"><a href="#import_dobs_string-460"><span class="linenos">460</span></a> <span class="n">deltad</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-461"><a href="#import_dobs_string-461"><span class="linenos">461</span></a> <span class="n">covd</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-462"><a href="#import_dobs_string-462"><span class="linenos">462</span></a> <span class="n">gradd</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-463"><a href="#import_dobs_string-463"><span class="linenos">463</span></a> <span class="n">names</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="import_dobs_string-464"><a href="#import_dobs_string-464"><span class="linenos">464</span></a> <span class="n">e_names</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="import_dobs_string-465"><a href="#import_dobs_string-465"><span class="linenos">465</span></a> <span class="n">enstags</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-466"><a href="#import_dobs_string-466"><span class="linenos">466</span></a> <span class="k">for</span> <span class="n">k</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">6</span><span class="p">,</span> <span class="nb">len</span><span class="p">(</span><span class="nb">list</span><span class="p">(</span><span class="n">dobs</span><span class="p">))):</span>
|
||||
</span><span id="import_dobs_string-467"><a href="#import_dobs_string-467"><span class="linenos">467</span></a> <span class="k">if</span> <span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"edata"</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-468"><a href="#import_dobs_string-468"><span class="linenos">468</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"enstag"</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-469"><a href="#import_dobs_string-469"><span class="linenos">469</span></a> <span class="n">ename</span> <span class="o">=</span> <span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
|
||||
</span><span id="import_dobs_string-470"><a href="#import_dobs_string-470"><span class="linenos">470</span></a> <span class="n">e_names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">ename</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-471"><a href="#import_dobs_string-471"><span class="linenos">471</span></a> <span class="n">_check</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"nr"</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-472"><a href="#import_dobs_string-472"><span class="linenos">472</span></a> <span class="n">R</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">text</span><span class="o">.</span><span class="n">strip</span><span class="p">())</span>
|
||||
</span><span id="import_dobs_string-473"><a href="#import_dobs_string-473"><span class="linenos">473</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">2</span><span class="p">,</span> <span class="mi">2</span> <span class="o">+</span> <span class="n">R</span><span class="p">):</span>
|
||||
</span><span id="import_dobs_string-474"><a href="#import_dobs_string-474"><span class="linenos">474</span></a> <span class="n">deltas</span><span class="p">,</span> <span class="n">rname</span><span class="p">,</span> <span class="n">idx</span> <span class="o">=</span> <span class="n">_import_rdata</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="n">i</span><span class="p">])</span>
|
||||
</span><span id="import_dobs_string-475"><a href="#import_dobs_string-475"><span class="linenos">475</span></a> <span class="k">if</span> <span class="n">separator_insertion</span> <span class="ow">is</span> <span class="kc">None</span> <span class="ow">or</span> <span class="kc">False</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-476"><a href="#import_dobs_string-476"><span class="linenos">476</span></a> <span class="k">pass</span>
|
||||
</span><span id="import_dobs_string-477"><a href="#import_dobs_string-477"><span class="linenos">477</span></a> <span class="k">elif</span> <span class="n">separator_insertion</span> <span class="ow">is</span> <span class="kc">True</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-478"><a href="#import_dobs_string-478"><span class="linenos">478</span></a> <span class="k">if</span> <span class="n">rname</span><span class="o">.</span><span class="n">startswith</span><span class="p">(</span><span class="n">ename</span><span class="p">):</span>
|
||||
</span><span id="import_dobs_string-479"><a href="#import_dobs_string-479"><span class="linenos">479</span></a> <span class="n">rname</span> <span class="o">=</span> <span class="n">rname</span><span class="p">[:</span><span class="nb">len</span><span class="p">(</span><span class="n">ename</span><span class="p">)]</span> <span class="o">+</span> <span class="s1">'|'</span> <span class="o">+</span> <span class="n">rname</span><span class="p">[</span><span class="nb">len</span><span class="p">(</span><span class="n">ename</span><span class="p">):]</span>
|
||||
</span><span id="import_dobs_string-480"><a href="#import_dobs_string-480"><span class="linenos">480</span></a> <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">,</span> <span class="nb">int</span><span class="p">):</span>
|
||||
</span><span id="import_dobs_string-481"><a href="#import_dobs_string-481"><span class="linenos">481</span></a> <span class="n">rname</span> <span class="o">=</span> <span class="n">rname</span><span class="p">[:</span><span class="n">separator_insertion</span><span class="p">]</span> <span class="o">+</span> <span class="s1">'|'</span> <span class="o">+</span> <span class="n">rname</span><span class="p">[</span><span class="n">separator_insertion</span><span class="p">:]</span>
|
||||
</span><span id="import_dobs_string-482"><a href="#import_dobs_string-482"><span class="linenos">482</span></a> <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">,</span> <span class="nb">str</span><span class="p">):</span>
|
||||
</span><span id="import_dobs_string-483"><a href="#import_dobs_string-483"><span class="linenos">483</span></a> <span class="n">rname</span> <span class="o">=</span> <span class="n">rname</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">,</span> <span class="s2">"|</span><span class="si">%s</span><span class="s2">"</span> <span class="o">%</span> <span class="p">(</span><span class="n">separator_insertion</span><span class="p">))</span>
|
||||
</span><span id="import_dobs_string-484"><a href="#import_dobs_string-484"><span class="linenos">484</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-485"><a href="#import_dobs_string-485"><span class="linenos">485</span></a> <span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="s2">"separator_insertion has to be string or int, is "</span><span class="p">,</span> <span class="nb">type</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">))</span>
|
||||
</span><span id="import_dobs_string-486"><a href="#import_dobs_string-486"><span class="linenos">486</span></a> <span class="k">if</span> <span class="s1">'|'</span> <span class="ow">in</span> <span class="n">rname</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-487"><a href="#import_dobs_string-487"><span class="linenos">487</span></a> <span class="n">new_ename</span> <span class="o">=</span> <span class="n">rname</span><span class="p">[:</span><span class="n">rname</span><span class="o">.</span><span class="n">index</span><span class="p">(</span><span class="s1">'|'</span><span class="p">)]</span>
|
||||
</span><span id="import_dobs_string-488"><a href="#import_dobs_string-488"><span class="linenos">488</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-489"><a href="#import_dobs_string-489"><span class="linenos">489</span></a> <span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="s2">"separator_insertion has to be string or int, is "</span><span class="p">,</span> <span class="nb">type</span><span class="p">(</span><span class="n">separator_insertion</span><span class="p">))</span>
|
||||
</span><span id="import_dobs_string-490"><a href="#import_dobs_string-490"><span class="linenos">490</span></a> <span class="k">if</span> <span class="s1">'|'</span> <span class="ow">in</span> <span class="n">rname</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-491"><a href="#import_dobs_string-491"><span class="linenos">491</span></a> <span class="n">new_ename</span> <span class="o">=</span> <span class="n">rname</span><span class="p">[:</span><span class="n">rname</span><span class="o">.</span><span class="n">index</span><span class="p">(</span><span class="s1">'|'</span><span class="p">)]</span>
|
||||
</span><span id="import_dobs_string-492"><a href="#import_dobs_string-492"><span class="linenos">492</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-493"><a href="#import_dobs_string-493"><span class="linenos">493</span></a> <span class="n">new_ename</span> <span class="o">=</span> <span class="n">ename</span>
|
||||
</span><span id="import_dobs_string-494"><a href="#import_dobs_string-494"><span class="linenos">494</span></a> <span class="n">enstags</span><span class="p">[</span><span class="n">new_ename</span><span class="p">]</span> <span class="o">=</span> <span class="n">ename</span>
|
||||
</span><span id="import_dobs_string-495"><a href="#import_dobs_string-495"><span class="linenos">495</span></a> <span class="n">idld</span><span class="p">[</span><span class="n">rname</span><span class="p">]</span> <span class="o">=</span> <span class="n">idx</span>
|
||||
</span><span id="import_dobs_string-496"><a href="#import_dobs_string-496"><span class="linenos">496</span></a> <span class="n">deltad</span><span class="p">[</span><span class="n">rname</span><span class="p">]</span> <span class="o">=</span> <span class="n">deltas</span>
|
||||
</span><span id="import_dobs_string-497"><a href="#import_dobs_string-497"><span class="linenos">497</span></a> <span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">rname</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-498"><a href="#import_dobs_string-498"><span class="linenos">498</span></a> <span class="k">elif</span> <span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"cdata"</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-499"><a href="#import_dobs_string-499"><span class="linenos">499</span></a> <span class="n">cname</span><span class="p">,</span> <span class="n">cov</span><span class="p">,</span> <span class="n">grad</span> <span class="o">=</span> <span class="n">_import_cdata</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">])</span>
|
||||
</span><span id="import_dobs_string-500"><a href="#import_dobs_string-500"><span class="linenos">500</span></a> <span class="n">covd</span><span class="p">[</span><span class="n">cname</span><span class="p">]</span> <span class="o">=</span> <span class="n">cov</span>
|
||||
</span><span id="import_dobs_string-501"><a href="#import_dobs_string-501"><span class="linenos">501</span></a> <span class="k">if</span> <span class="n">grad</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span> <span class="o">==</span> <span class="mi">1</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-502"><a href="#import_dobs_string-502"><span class="linenos">502</span></a> <span class="n">gradd</span><span class="p">[</span><span class="n">cname</span><span class="p">]</span> <span class="o">=</span> <span class="p">[</span><span class="n">grad</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">mean</span><span class="p">))]</span>
|
||||
</span><span id="import_dobs_string-503"><a href="#import_dobs_string-503"><span class="linenos">503</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-504"><a href="#import_dobs_string-504"><span class="linenos">504</span></a> <span class="n">gradd</span><span class="p">[</span><span class="n">cname</span><span class="p">]</span> <span class="o">=</span> <span class="n">grad</span><span class="o">.</span><span class="n">T</span>
|
||||
</span><span id="import_dobs_string-505"><a href="#import_dobs_string-505"><span class="linenos">505</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-506"><a href="#import_dobs_string-506"><span class="linenos">506</span></a> <span class="n">_check</span><span class="p">(</span><span class="kc">False</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-507"><a href="#import_dobs_string-507"><span class="linenos">507</span></a> <span class="n">names</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="nb">set</span><span class="p">(</span><span class="n">names</span><span class="p">))</span>
|
||||
</span><span id="import_dobs_string-508"><a href="#import_dobs_string-508"><span class="linenos">508</span></a>
|
||||
</span><span id="import_dobs_string-509"><a href="#import_dobs_string-509"><span class="linenos">509</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">names</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-510"><a href="#import_dobs_string-510"><span class="linenos">510</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">])):</span>
|
||||
</span><span id="import_dobs_string-511"><a href="#import_dobs_string-511"><span class="linenos">511</span></a> <span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">]</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">array</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])</span> <span class="o">+</span> <span class="n">mean</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-489"><a href="#import_dobs_string-489"><span class="linenos">489</span></a> <span class="n">new_ename</span> <span class="o">=</span> <span class="n">ename</span>
|
||||
</span><span id="import_dobs_string-490"><a href="#import_dobs_string-490"><span class="linenos">490</span></a> <span class="n">enstags</span><span class="p">[</span><span class="n">new_ename</span><span class="p">]</span> <span class="o">=</span> <span class="n">ename</span>
|
||||
</span><span id="import_dobs_string-491"><a href="#import_dobs_string-491"><span class="linenos">491</span></a> <span class="n">idld</span><span class="p">[</span><span class="n">rname</span><span class="p">]</span> <span class="o">=</span> <span class="n">idx</span>
|
||||
</span><span id="import_dobs_string-492"><a href="#import_dobs_string-492"><span class="linenos">492</span></a> <span class="n">deltad</span><span class="p">[</span><span class="n">rname</span><span class="p">]</span> <span class="o">=</span> <span class="n">deltas</span>
|
||||
</span><span id="import_dobs_string-493"><a href="#import_dobs_string-493"><span class="linenos">493</span></a> <span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">rname</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-494"><a href="#import_dobs_string-494"><span class="linenos">494</span></a> <span class="k">elif</span> <span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s2">"cdata"</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-495"><a href="#import_dobs_string-495"><span class="linenos">495</span></a> <span class="n">cname</span><span class="p">,</span> <span class="n">cov</span><span class="p">,</span> <span class="n">grad</span> <span class="o">=</span> <span class="n">_import_cdata</span><span class="p">(</span><span class="n">dobs</span><span class="p">[</span><span class="n">k</span><span class="p">])</span>
|
||||
</span><span id="import_dobs_string-496"><a href="#import_dobs_string-496"><span class="linenos">496</span></a> <span class="n">covd</span><span class="p">[</span><span class="n">cname</span><span class="p">]</span> <span class="o">=</span> <span class="n">cov</span>
|
||||
</span><span id="import_dobs_string-497"><a href="#import_dobs_string-497"><span class="linenos">497</span></a> <span class="k">if</span> <span class="n">grad</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span> <span class="o">==</span> <span class="mi">1</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-498"><a href="#import_dobs_string-498"><span class="linenos">498</span></a> <span class="n">gradd</span><span class="p">[</span><span class="n">cname</span><span class="p">]</span> <span class="o">=</span> <span class="p">[</span><span class="n">grad</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">mean</span><span class="p">))]</span>
|
||||
</span><span id="import_dobs_string-499"><a href="#import_dobs_string-499"><span class="linenos">499</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-500"><a href="#import_dobs_string-500"><span class="linenos">500</span></a> <span class="n">gradd</span><span class="p">[</span><span class="n">cname</span><span class="p">]</span> <span class="o">=</span> <span class="n">grad</span><span class="o">.</span><span class="n">T</span>
|
||||
</span><span id="import_dobs_string-501"><a href="#import_dobs_string-501"><span class="linenos">501</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-502"><a href="#import_dobs_string-502"><span class="linenos">502</span></a> <span class="n">_check</span><span class="p">(</span><span class="kc">False</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-503"><a href="#import_dobs_string-503"><span class="linenos">503</span></a> <span class="n">names</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="nb">set</span><span class="p">(</span><span class="n">names</span><span class="p">))</span>
|
||||
</span><span id="import_dobs_string-504"><a href="#import_dobs_string-504"><span class="linenos">504</span></a>
|
||||
</span><span id="import_dobs_string-505"><a href="#import_dobs_string-505"><span class="linenos">505</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">names</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-506"><a href="#import_dobs_string-506"><span class="linenos">506</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">])):</span>
|
||||
</span><span id="import_dobs_string-507"><a href="#import_dobs_string-507"><span class="linenos">507</span></a> <span class="n">tmp</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">zeros_like</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])</span>
|
||||
</span><span id="import_dobs_string-508"><a href="#import_dobs_string-508"><span class="linenos">508</span></a> <span class="k">for</span> <span class="n">j</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])):</span>
|
||||
</span><span id="import_dobs_string-509"><a href="#import_dobs_string-509"><span class="linenos">509</span></a> <span class="k">if</span> <span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">][</span><span class="n">j</span><span class="p">]</span> <span class="o">!=</span> <span class="mf">0.</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-510"><a href="#import_dobs_string-510"><span class="linenos">510</span></a> <span class="n">tmp</span><span class="p">[</span><span class="n">j</span><span class="p">]</span> <span class="o">=</span> <span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">][</span><span class="n">j</span><span class="p">]</span> <span class="o">+</span> <span class="n">mean</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-511"><a href="#import_dobs_string-511"><span class="linenos">511</span></a> <span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">]</span> <span class="o">=</span> <span class="n">tmp</span>
|
||||
</span><span id="import_dobs_string-512"><a href="#import_dobs_string-512"><span class="linenos">512</span></a>
|
||||
</span><span id="import_dobs_string-513"><a href="#import_dobs_string-513"><span class="linenos">513</span></a> <span class="n">res</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="import_dobs_string-514"><a href="#import_dobs_string-514"><span class="linenos">514</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">mean</span><span class="p">)):</span>
|
||||
|
|
@ -1522,56 +1522,59 @@ Imported data and meta-data</li>
|
|||
</span><span id="import_dobs_string-517"><a href="#import_dobs_string-517"><span class="linenos">517</span></a> <span class="n">obs_names</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="import_dobs_string-518"><a href="#import_dobs_string-518"><span class="linenos">518</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">names</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-519"><a href="#import_dobs_string-519"><span class="linenos">519</span></a> <span class="n">h</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">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])</span>
|
||||
</span><span id="import_dobs_string-520"><a href="#import_dobs_string-520"><span class="linenos">520</span></a> <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">h</span><span class="p">)</span> <span class="o">==</span> <span class="mi">1</span> <span class="ow">and</span> <span class="n">np</span><span class="o">.</span><span class="n">all</span><span class="p">(</span><span class="n">h</span> <span class="o">==</span> <span class="n">mean</span><span class="p">[</span><span class="n">i</span><span class="p">])</span> <span class="ow">and</span> <span class="n">noempty</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-520"><a href="#import_dobs_string-520"><span class="linenos">520</span></a> <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">h</span><span class="p">)</span> <span class="o">==</span> <span class="mi">1</span> <span class="ow">and</span> <span class="n">np</span><span class="o">.</span><span class="n">all</span><span class="p">(</span><span class="n">h</span> <span class="o">==</span> <span class="n">mean</span><span class="p">[</span><span class="n">i</span><span class="p">]):</span>
|
||||
</span><span id="import_dobs_string-521"><a href="#import_dobs_string-521"><span class="linenos">521</span></a> <span class="k">continue</span>
|
||||
</span><span id="import_dobs_string-522"><a href="#import_dobs_string-522"><span class="linenos">522</span></a> <span class="n">deltas</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])</span>
|
||||
</span><span id="import_dobs_string-523"><a href="#import_dobs_string-523"><span class="linenos">523</span></a> <span class="n">obs_names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">name</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-524"><a href="#import_dobs_string-524"><span class="linenos">524</span></a> <span class="n">idl</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">idld</span><span class="p">[</span><span class="n">name</span><span class="p">])</span>
|
||||
</span><span id="import_dobs_string-525"><a href="#import_dobs_string-525"><span class="linenos">525</span></a> <span class="n">res</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">deltas</span><span class="p">,</span> <span class="n">obs_names</span><span class="p">,</span> <span class="n">idl</span><span class="o">=</span><span class="n">idl</span><span class="p">))</span>
|
||||
</span><span id="import_dobs_string-526"><a href="#import_dobs_string-526"><span class="linenos">526</span></a> <span class="n">res</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">_value</span> <span class="o">=</span> <span class="n">mean</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-527"><a href="#import_dobs_string-527"><span class="linenos">527</span></a> <span class="n">_check</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">e_names</span><span class="p">)</span> <span class="o">==</span> <span class="n">ne</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-528"><a href="#import_dobs_string-528"><span class="linenos">528</span></a>
|
||||
</span><span id="import_dobs_string-529"><a href="#import_dobs_string-529"><span class="linenos">529</span></a> <span class="n">cnames</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="n">covd</span><span class="o">.</span><span class="n">keys</span><span class="p">())</span>
|
||||
</span><span id="import_dobs_string-530"><a href="#import_dobs_string-530"><span class="linenos">530</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">res</span><span class="p">)):</span>
|
||||
</span><span id="import_dobs_string-531"><a href="#import_dobs_string-531"><span class="linenos">531</span></a> <span class="n">new_covobs</span> <span class="o">=</span> <span class="p">{</span><span class="n">name</span><span class="p">:</span> <span class="n">Covobs</span><span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="n">covd</span><span class="p">[</span><span class="n">name</span><span class="p">],</span> <span class="n">name</span><span class="p">,</span> <span class="n">grad</span><span class="o">=</span><span class="n">gradd</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])</span> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">cnames</span><span class="p">}</span>
|
||||
</span><span id="import_dobs_string-532"><a href="#import_dobs_string-532"><span class="linenos">532</span></a> <span class="k">if</span> <span class="n">noempty</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-533"><a href="#import_dobs_string-533"><span class="linenos">533</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">cnames</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-534"><a href="#import_dobs_string-534"><span class="linenos">534</span></a> <span class="k">if</span> <span class="n">np</span><span class="o">.</span><span class="n">all</span><span class="p">(</span><span class="n">new_covobs</span><span class="p">[</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">grad</span> <span class="o">==</span> <span class="mi">0</span><span class="p">):</span>
|
||||
</span><span id="import_dobs_string-535"><a href="#import_dobs_string-535"><span class="linenos">535</span></a> <span class="k">del</span> <span class="n">new_covobs</span><span class="p">[</span><span class="n">name</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-536"><a href="#import_dobs_string-536"><span class="linenos">536</span></a> <span class="n">cnames_loc</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="n">new_covobs</span><span class="o">.</span><span class="n">keys</span><span class="p">())</span>
|
||||
</span><span id="import_dobs_string-537"><a href="#import_dobs_string-537"><span class="linenos">537</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-538"><a href="#import_dobs_string-538"><span class="linenos">538</span></a> <span class="n">cnames_loc</span> <span class="o">=</span> <span class="n">cnames</span>
|
||||
</span><span id="import_dobs_string-539"><a href="#import_dobs_string-539"><span class="linenos">539</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">cnames_loc</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-540"><a href="#import_dobs_string-540"><span class="linenos">540</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">name</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-541"><a href="#import_dobs_string-541"><span class="linenos">541</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="mi">1</span>
|
||||
</span><span id="import_dobs_string-542"><a href="#import_dobs_string-542"><span class="linenos">542</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">idl</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="import_dobs_string-543"><a href="#import_dobs_string-543"><span class="linenos">543</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">_covobs</span> <span class="o">=</span> <span class="n">new_covobs</span>
|
||||
</span><span id="import_dobs_string-544"><a href="#import_dobs_string-544"><span class="linenos">544</span></a>
|
||||
</span><span id="import_dobs_string-545"><a href="#import_dobs_string-545"><span class="linenos">545</span></a> <span class="k">if</span> <span class="n">symbol</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-546"><a href="#import_dobs_string-546"><span class="linenos">546</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">res</span><span class="p">)):</span>
|
||||
</span><span id="import_dobs_string-547"><a href="#import_dobs_string-547"><span class="linenos">547</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">=</span> <span class="n">symbol</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-548"><a href="#import_dobs_string-548"><span class="linenos">548</span></a> <span class="k">if</span> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'None'</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-549"><a href="#import_dobs_string-549"><span class="linenos">549</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">=</span> <span class="kc">None</span>
|
||||
</span><span id="import_dobs_string-550"><a href="#import_dobs_string-550"><span class="linenos">550</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">noempty</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-551"><a href="#import_dobs_string-551"><span class="linenos">551</span></a> <span class="n">_check</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">res</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">covobs</span><span class="o">.</span><span class="n">keys</span><span class="p">())</span> <span class="o">==</span> <span class="n">nc</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-552"><a href="#import_dobs_string-552"><span class="linenos">552</span></a> <span class="k">if</span> <span class="n">full_output</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-553"><a href="#import_dobs_string-553"><span class="linenos">553</span></a> <span class="n">retd</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-554"><a href="#import_dobs_string-554"><span class="linenos">554</span></a> <span class="n">tool</span> <span class="o">=</span> <span class="n">file_origin</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s1">'tool'</span><span class="p">,</span> <span class="kc">None</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-555"><a href="#import_dobs_string-555"><span class="linenos">555</span></a> <span class="k">if</span> <span class="n">tool</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-556"><a href="#import_dobs_string-556"><span class="linenos">556</span></a> <span class="n">program</span> <span class="o">=</span> <span class="n">tool</span><span class="p">[</span><span class="s1">'name'</span><span class="p">]</span> <span class="o">+</span> <span class="s1">' '</span> <span class="o">+</span> <span class="n">tool</span><span class="p">[</span><span class="s1">'version'</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-557"><a href="#import_dobs_string-557"><span class="linenos">557</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-558"><a href="#import_dobs_string-558"><span class="linenos">558</span></a> <span class="n">program</span> <span class="o">=</span> <span class="s1">''</span>
|
||||
</span><span id="import_dobs_string-559"><a href="#import_dobs_string-559"><span class="linenos">559</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'program'</span><span class="p">]</span> <span class="o">=</span> <span class="n">program</span>
|
||||
</span><span id="import_dobs_string-560"><a href="#import_dobs_string-560"><span class="linenos">560</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'version'</span><span class="p">]</span> <span class="o">=</span> <span class="n">version</span>
|
||||
</span><span id="import_dobs_string-561"><a href="#import_dobs_string-561"><span class="linenos">561</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'who'</span><span class="p">]</span> <span class="o">=</span> <span class="n">file_origin</span><span class="p">[</span><span class="s1">'who'</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-562"><a href="#import_dobs_string-562"><span class="linenos">562</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'date'</span><span class="p">]</span> <span class="o">=</span> <span class="n">file_origin</span><span class="p">[</span><span class="s1">'date'</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-563"><a href="#import_dobs_string-563"><span class="linenos">563</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'host'</span><span class="p">]</span> <span class="o">=</span> <span class="n">file_origin</span><span class="p">[</span><span class="s1">'host'</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-564"><a href="#import_dobs_string-564"><span class="linenos">564</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'description'</span><span class="p">]</span> <span class="o">=</span> <span class="n">descriptiond</span>
|
||||
</span><span id="import_dobs_string-565"><a href="#import_dobs_string-565"><span class="linenos">565</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'enstags'</span><span class="p">]</span> <span class="o">=</span> <span class="n">enstags</span>
|
||||
</span><span id="import_dobs_string-566"><a href="#import_dobs_string-566"><span class="linenos">566</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'obsdata'</span><span class="p">]</span> <span class="o">=</span> <span class="n">res</span>
|
||||
</span><span id="import_dobs_string-567"><a href="#import_dobs_string-567"><span class="linenos">567</span></a> <span class="k">return</span> <span class="n">retd</span>
|
||||
</span><span id="import_dobs_string-568"><a href="#import_dobs_string-568"><span class="linenos">568</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-569"><a href="#import_dobs_string-569"><span class="linenos">569</span></a> <span class="k">return</span> <span class="n">res</span>
|
||||
</span><span id="import_dobs_string-522"><a href="#import_dobs_string-522"><span class="linenos">522</span></a> <span class="n">repdeltas</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="import_dobs_string-523"><a href="#import_dobs_string-523"><span class="linenos">523</span></a> <span class="n">repidl</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="import_dobs_string-524"><a href="#import_dobs_string-524"><span class="linenos">524</span></a> <span class="k">for</span> <span class="n">j</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])):</span>
|
||||
</span><span id="import_dobs_string-525"><a href="#import_dobs_string-525"><span class="linenos">525</span></a> <span class="k">if</span> <span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">][</span><span class="n">j</span><span class="p">]</span> <span class="o">!=</span> <span class="mf">0.</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-526"><a href="#import_dobs_string-526"><span class="linenos">526</span></a> <span class="n">repdeltas</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">deltad</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">][</span><span class="n">j</span><span class="p">])</span>
|
||||
</span><span id="import_dobs_string-527"><a href="#import_dobs_string-527"><span class="linenos">527</span></a> <span class="n">repidl</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">idld</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">j</span><span class="p">])</span>
|
||||
</span><span id="import_dobs_string-528"><a href="#import_dobs_string-528"><span class="linenos">528</span></a> <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">repdeltas</span><span class="p">)</span> <span class="o">></span> <span class="mi">0</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-529"><a href="#import_dobs_string-529"><span class="linenos">529</span></a> <span class="n">obs_names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">name</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-530"><a href="#import_dobs_string-530"><span class="linenos">530</span></a> <span class="n">deltas</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">repdeltas</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-531"><a href="#import_dobs_string-531"><span class="linenos">531</span></a> <span class="n">idl</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">repidl</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-532"><a href="#import_dobs_string-532"><span class="linenos">532</span></a>
|
||||
</span><span id="import_dobs_string-533"><a href="#import_dobs_string-533"><span class="linenos">533</span></a> <span class="n">res</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">deltas</span><span class="p">,</span> <span class="n">obs_names</span><span class="p">,</span> <span class="n">idl</span><span class="o">=</span><span class="n">idl</span><span class="p">))</span>
|
||||
</span><span id="import_dobs_string-534"><a href="#import_dobs_string-534"><span class="linenos">534</span></a> <span class="n">res</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">_value</span> <span class="o">=</span> <span class="n">mean</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-535"><a href="#import_dobs_string-535"><span class="linenos">535</span></a> <span class="n">_check</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">e_names</span><span class="p">)</span> <span class="o">==</span> <span class="n">ne</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-536"><a href="#import_dobs_string-536"><span class="linenos">536</span></a>
|
||||
</span><span id="import_dobs_string-537"><a href="#import_dobs_string-537"><span class="linenos">537</span></a> <span class="n">cnames</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="n">covd</span><span class="o">.</span><span class="n">keys</span><span class="p">())</span>
|
||||
</span><span id="import_dobs_string-538"><a href="#import_dobs_string-538"><span class="linenos">538</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">res</span><span class="p">)):</span>
|
||||
</span><span id="import_dobs_string-539"><a href="#import_dobs_string-539"><span class="linenos">539</span></a> <span class="n">new_covobs</span> <span class="o">=</span> <span class="p">{</span><span class="n">name</span><span class="p">:</span> <span class="n">Covobs</span><span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="n">covd</span><span class="p">[</span><span class="n">name</span><span class="p">],</span> <span class="n">name</span><span class="p">,</span> <span class="n">grad</span><span class="o">=</span><span class="n">gradd</span><span class="p">[</span><span class="n">name</span><span class="p">][</span><span class="n">i</span><span class="p">])</span> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">cnames</span><span class="p">}</span>
|
||||
</span><span id="import_dobs_string-540"><a href="#import_dobs_string-540"><span class="linenos">540</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">cnames</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-541"><a href="#import_dobs_string-541"><span class="linenos">541</span></a> <span class="k">if</span> <span class="n">np</span><span class="o">.</span><span class="n">all</span><span class="p">(</span><span class="n">new_covobs</span><span class="p">[</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">grad</span> <span class="o">==</span> <span class="mi">0</span><span class="p">):</span>
|
||||
</span><span id="import_dobs_string-542"><a href="#import_dobs_string-542"><span class="linenos">542</span></a> <span class="k">del</span> <span class="n">new_covobs</span><span class="p">[</span><span class="n">name</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-543"><a href="#import_dobs_string-543"><span class="linenos">543</span></a> <span class="n">cnames_loc</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="n">new_covobs</span><span class="o">.</span><span class="n">keys</span><span class="p">())</span>
|
||||
</span><span id="import_dobs_string-544"><a href="#import_dobs_string-544"><span class="linenos">544</span></a> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">cnames_loc</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-545"><a href="#import_dobs_string-545"><span class="linenos">545</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">name</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-546"><a href="#import_dobs_string-546"><span class="linenos">546</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="mi">1</span>
|
||||
</span><span id="import_dobs_string-547"><a href="#import_dobs_string-547"><span class="linenos">547</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">idl</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
||||
</span><span id="import_dobs_string-548"><a href="#import_dobs_string-548"><span class="linenos">548</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">_covobs</span> <span class="o">=</span> <span class="n">new_covobs</span>
|
||||
</span><span id="import_dobs_string-549"><a href="#import_dobs_string-549"><span class="linenos">549</span></a>
|
||||
</span><span id="import_dobs_string-550"><a href="#import_dobs_string-550"><span class="linenos">550</span></a> <span class="k">if</span> <span class="n">symbol</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-551"><a href="#import_dobs_string-551"><span class="linenos">551</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">res</span><span class="p">)):</span>
|
||||
</span><span id="import_dobs_string-552"><a href="#import_dobs_string-552"><span class="linenos">552</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">=</span> <span class="n">symbol</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-553"><a href="#import_dobs_string-553"><span class="linenos">553</span></a> <span class="k">if</span> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">==</span> <span class="s1">'None'</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-554"><a href="#import_dobs_string-554"><span class="linenos">554</span></a> <span class="n">res</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">tag</span> <span class="o">=</span> <span class="kc">None</span>
|
||||
</span><span id="import_dobs_string-555"><a href="#import_dobs_string-555"><span class="linenos">555</span></a> <span class="k">if</span> <span class="n">full_output</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-556"><a href="#import_dobs_string-556"><span class="linenos">556</span></a> <span class="n">retd</span> <span class="o">=</span> <span class="p">{}</span>
|
||||
</span><span id="import_dobs_string-557"><a href="#import_dobs_string-557"><span class="linenos">557</span></a> <span class="n">tool</span> <span class="o">=</span> <span class="n">file_origin</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s1">'tool'</span><span class="p">,</span> <span class="kc">None</span><span class="p">)</span>
|
||||
</span><span id="import_dobs_string-558"><a href="#import_dobs_string-558"><span class="linenos">558</span></a> <span class="k">if</span> <span class="n">tool</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-559"><a href="#import_dobs_string-559"><span class="linenos">559</span></a> <span class="n">program</span> <span class="o">=</span> <span class="n">tool</span><span class="p">[</span><span class="s1">'name'</span><span class="p">]</span> <span class="o">+</span> <span class="s1">' '</span> <span class="o">+</span> <span class="n">tool</span><span class="p">[</span><span class="s1">'version'</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-560"><a href="#import_dobs_string-560"><span class="linenos">560</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-561"><a href="#import_dobs_string-561"><span class="linenos">561</span></a> <span class="n">program</span> <span class="o">=</span> <span class="s1">''</span>
|
||||
</span><span id="import_dobs_string-562"><a href="#import_dobs_string-562"><span class="linenos">562</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'program'</span><span class="p">]</span> <span class="o">=</span> <span class="n">program</span>
|
||||
</span><span id="import_dobs_string-563"><a href="#import_dobs_string-563"><span class="linenos">563</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'version'</span><span class="p">]</span> <span class="o">=</span> <span class="n">version</span>
|
||||
</span><span id="import_dobs_string-564"><a href="#import_dobs_string-564"><span class="linenos">564</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'who'</span><span class="p">]</span> <span class="o">=</span> <span class="n">file_origin</span><span class="p">[</span><span class="s1">'who'</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-565"><a href="#import_dobs_string-565"><span class="linenos">565</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'date'</span><span class="p">]</span> <span class="o">=</span> <span class="n">file_origin</span><span class="p">[</span><span class="s1">'date'</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-566"><a href="#import_dobs_string-566"><span class="linenos">566</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'host'</span><span class="p">]</span> <span class="o">=</span> <span class="n">file_origin</span><span class="p">[</span><span class="s1">'host'</span><span class="p">]</span>
|
||||
</span><span id="import_dobs_string-567"><a href="#import_dobs_string-567"><span class="linenos">567</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'description'</span><span class="p">]</span> <span class="o">=</span> <span class="n">descriptiond</span>
|
||||
</span><span id="import_dobs_string-568"><a href="#import_dobs_string-568"><span class="linenos">568</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'enstags'</span><span class="p">]</span> <span class="o">=</span> <span class="n">enstags</span>
|
||||
</span><span id="import_dobs_string-569"><a href="#import_dobs_string-569"><span class="linenos">569</span></a> <span class="n">retd</span><span class="p">[</span><span class="s1">'obsdata'</span><span class="p">]</span> <span class="o">=</span> <span class="n">res</span>
|
||||
</span><span id="import_dobs_string-570"><a href="#import_dobs_string-570"><span class="linenos">570</span></a> <span class="k">return</span> <span class="n">retd</span>
|
||||
</span><span id="import_dobs_string-571"><a href="#import_dobs_string-571"><span class="linenos">571</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="import_dobs_string-572"><a href="#import_dobs_string-572"><span class="linenos">572</span></a> <span class="k">return</span> <span class="n">res</span>
|
||||
</span></pre></div>
|
||||
|
||||
|
||||
|
|
@ -1584,9 +1587,6 @@ Imported data and meta-data</li>
|
|||
<ul>
|
||||
<li><strong>content</strong> (str):
|
||||
XML string containing the data</li>
|
||||
<li><strong>noemtpy</strong> (bool):
|
||||
If True, ensembles with no contribution to the Obs are not included.
|
||||
If False, ensembles are included as written in the file, possibly with vanishing entries.</li>
|
||||
<li><strong>full_output</strong> (bool):
|
||||
If True, a dict containing auxiliary information and the data is returned.
|
||||
If False, only the data is returned as list.</li>
|
||||
|
|
@ -1617,24 +1617,21 @@ Imported data and meta-data</li>
|
|||
<div class="attr function">
|
||||
|
||||
<span class="def">def</span>
|
||||
<span class="name">read_dobs</span><span class="signature pdoc-code multiline">(<span class="param"> <span class="n">fname</span>,</span><span class="param"> <span class="n">noempty</span><span class="o">=</span><span class="kc">False</span>,</span><span class="param"> <span class="n">full_output</span><span class="o">=</span><span class="kc">False</span>,</span><span class="param"> <span class="n">gz</span><span class="o">=</span><span class="kc">True</span>,</span><span class="param"> <span class="n">separator_insertion</span><span class="o">=</span><span class="kc">True</span></span><span class="return-annotation">):</span></span>
|
||||
<span class="name">read_dobs</span><span class="signature pdoc-code condensed">(<span class="param"><span class="n">fname</span>, </span><span class="param"><span class="n">full_output</span><span class="o">=</span><span class="kc">False</span>, </span><span class="param"><span class="n">gz</span><span class="o">=</span><span class="kc">True</span>, </span><span class="param"><span class="n">separator_insertion</span><span class="o">=</span><span class="kc">True</span></span><span class="return-annotation">):</span></span>
|
||||
|
||||
<label class="view-source-button" for="read_dobs-view-source"><span>View Source</span></label>
|
||||
|
||||
</div>
|
||||
<a class="headerlink" href="#read_dobs"></a>
|
||||
<div class="pdoc-code codehilite"><pre><span></span><span id="read_dobs-572"><a href="#read_dobs-572"><span class="linenos">572</span></a><span class="k">def</span> <span class="nf">read_dobs</span><span class="p">(</span><span class="n">fname</span><span class="p">,</span> <span class="n">noempty</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">full_output</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">gz</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">separator_insertion</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>
|
||||
</span><span id="read_dobs-573"><a href="#read_dobs-573"><span class="linenos">573</span></a><span class="w"> </span><span class="sd">"""Import a list of Obs from an xml.gz file in the Zeuthen dobs format.</span>
|
||||
</span><span id="read_dobs-574"><a href="#read_dobs-574"><span class="linenos">574</span></a>
|
||||
</span><span id="read_dobs-575"><a href="#read_dobs-575"><span class="linenos">575</span></a><span class="sd"> Tags are not written or recovered automatically.</span>
|
||||
</span><span id="read_dobs-576"><a href="#read_dobs-576"><span class="linenos">576</span></a>
|
||||
</span><span id="read_dobs-577"><a href="#read_dobs-577"><span class="linenos">577</span></a><span class="sd"> Parameters</span>
|
||||
</span><span id="read_dobs-578"><a href="#read_dobs-578"><span class="linenos">578</span></a><span class="sd"> ----------</span>
|
||||
</span><span id="read_dobs-579"><a href="#read_dobs-579"><span class="linenos">579</span></a><span class="sd"> fname : str</span>
|
||||
</span><span id="read_dobs-580"><a href="#read_dobs-580"><span class="linenos">580</span></a><span class="sd"> Filename of the input file.</span>
|
||||
</span><span id="read_dobs-581"><a href="#read_dobs-581"><span class="linenos">581</span></a><span class="sd"> noemtpy : bool</span>
|
||||
</span><span id="read_dobs-582"><a href="#read_dobs-582"><span class="linenos">582</span></a><span class="sd"> If True, ensembles with no contribution to the Obs are not included.</span>
|
||||
</span><span id="read_dobs-583"><a href="#read_dobs-583"><span class="linenos">583</span></a><span class="sd"> If False, ensembles are included as written in the file.</span>
|
||||
<div class="pdoc-code codehilite"><pre><span></span><span id="read_dobs-575"><a href="#read_dobs-575"><span class="linenos">575</span></a><span class="k">def</span> <span class="nf">read_dobs</span><span class="p">(</span><span class="n">fname</span><span class="p">,</span> <span class="n">full_output</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">gz</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">separator_insertion</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>
|
||||
</span><span id="read_dobs-576"><a href="#read_dobs-576"><span class="linenos">576</span></a><span class="w"> </span><span class="sd">"""Import a list of Obs from an xml.gz file in the Zeuthen dobs format.</span>
|
||||
</span><span id="read_dobs-577"><a href="#read_dobs-577"><span class="linenos">577</span></a>
|
||||
</span><span id="read_dobs-578"><a href="#read_dobs-578"><span class="linenos">578</span></a><span class="sd"> Tags are not written or recovered automatically.</span>
|
||||
</span><span id="read_dobs-579"><a href="#read_dobs-579"><span class="linenos">579</span></a>
|
||||
</span><span id="read_dobs-580"><a href="#read_dobs-580"><span class="linenos">580</span></a><span class="sd"> Parameters</span>
|
||||
</span><span id="read_dobs-581"><a href="#read_dobs-581"><span class="linenos">581</span></a><span class="sd"> ----------</span>
|
||||
</span><span id="read_dobs-582"><a href="#read_dobs-582"><span class="linenos">582</span></a><span class="sd"> fname : str</span>
|
||||
</span><span id="read_dobs-583"><a href="#read_dobs-583"><span class="linenos">583</span></a><span class="sd"> Filename of the input file.</span>
|
||||
</span><span id="read_dobs-584"><a href="#read_dobs-584"><span class="linenos">584</span></a><span class="sd"> full_output : bool</span>
|
||||
</span><span id="read_dobs-585"><a href="#read_dobs-585"><span class="linenos">585</span></a><span class="sd"> If True, a dict containing auxiliary information and the data is returned.</span>
|
||||
</span><span id="read_dobs-586"><a href="#read_dobs-586"><span class="linenos">586</span></a><span class="sd"> If False, only the data is returned as list.</span>
|
||||
|
|
@ -1670,7 +1667,7 @@ Imported data and meta-data</li>
|
|||
</span><span id="read_dobs-616"><a href="#read_dobs-616"><span class="linenos">616</span></a> <span class="k">with</span> <span class="nb">open</span><span class="p">(</span><span class="n">fname</span><span class="p">,</span> <span class="s1">'r'</span><span class="p">)</span> <span class="k">as</span> <span class="n">fin</span><span class="p">:</span>
|
||||
</span><span id="read_dobs-617"><a href="#read_dobs-617"><span class="linenos">617</span></a> <span class="n">content</span> <span class="o">=</span> <span class="n">fin</span><span class="o">.</span><span class="n">read</span><span class="p">()</span>
|
||||
</span><span id="read_dobs-618"><a href="#read_dobs-618"><span class="linenos">618</span></a>
|
||||
</span><span id="read_dobs-619"><a href="#read_dobs-619"><span class="linenos">619</span></a> <span class="k">return</span> <span class="n">import_dobs_string</span><span class="p">(</span><span class="n">content</span><span class="p">,</span> <span class="n">noempty</span><span class="p">,</span> <span class="n">full_output</span><span class="p">,</span> <span class="n">separator_insertion</span><span class="o">=</span><span class="n">separator_insertion</span><span class="p">)</span>
|
||||
</span><span id="read_dobs-619"><a href="#read_dobs-619"><span class="linenos">619</span></a> <span class="k">return</span> <span class="n">import_dobs_string</span><span class="p">(</span><span class="n">content</span><span class="p">,</span> <span class="n">full_output</span><span class="p">,</span> <span class="n">separator_insertion</span><span class="o">=</span><span class="n">separator_insertion</span><span class="p">)</span>
|
||||
</span></pre></div>
|
||||
|
||||
|
||||
|
|
@ -1683,9 +1680,6 @@ Imported data and meta-data</li>
|
|||
<ul>
|
||||
<li><strong>fname</strong> (str):
|
||||
Filename of the input file.</li>
|
||||
<li><strong>noemtpy</strong> (bool):
|
||||
If True, ensembles with no contribution to the Obs are not included.
|
||||
If False, ensembles are included as written in the file.</li>
|
||||
<li><strong>full_output</strong> (bool):
|
||||
If True, a dict containing auxiliary information and the data is returned.
|
||||
If False, only the data is returned as list.</li>
|
||||
|
|
@ -1829,7 +1823,7 @@ Imported data and meta-data</li>
|
|||
</span><span id="create_dobs_string-783"><a href="#create_dobs_string-783"><span class="linenos">783</span></a> <span class="n">o</span> <span class="o">=</span> <span class="n">obsl</span><span class="p">[</span><span class="n">oi</span><span class="p">]</span>
|
||||
</span><span id="create_dobs_string-784"><a href="#create_dobs_string-784"><span class="linenos">784</span></a> <span class="k">if</span> <span class="n">repname</span> <span class="ow">in</span> <span class="n">o</span><span class="o">.</span><span class="n">idl</span><span class="p">:</span>
|
||||
</span><span id="create_dobs_string-785"><a href="#create_dobs_string-785"><span class="linenos">785</span></a> <span class="k">if</span> <span class="n">counters</span><span class="p">[</span><span class="n">oi</span><span class="p">]</span> <span class="o"><</span> <span class="mi">0</span><span class="p">:</span>
|
||||
</span><span id="create_dobs_string-786"><a href="#create_dobs_string-786"><span class="linenos">786</span></a> <span class="n">num</span> <span class="o">=</span> <span class="n">offsets</span><span class="p">[</span><span class="n">oi</span><span class="p">]</span>
|
||||
</span><span id="create_dobs_string-786"><a href="#create_dobs_string-786"><span class="linenos">786</span></a> <span class="n">num</span> <span class="o">=</span> <span class="mi">0</span>
|
||||
</span><span id="create_dobs_string-787"><a href="#create_dobs_string-787"><span class="linenos">787</span></a> <span class="k">if</span> <span class="n">num</span> <span class="o">==</span> <span class="mi">0</span><span class="p">:</span>
|
||||
</span><span id="create_dobs_string-788"><a href="#create_dobs_string-788"><span class="linenos">788</span></a> <span class="n">data</span> <span class="o">+=</span> <span class="s1">'0 '</span>
|
||||
</span><span id="create_dobs_string-789"><a href="#create_dobs_string-789"><span class="linenos">789</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
|
|
@ -1845,7 +1839,7 @@ Imported data and meta-data</li>
|
|||
</span><span id="create_dobs_string-799"><a href="#create_dobs_string-799"><span class="linenos">799</span></a> <span class="k">if</span> <span class="n">counters</span><span class="p">[</span><span class="n">oi</span><span class="p">]</span> <span class="o">>=</span> <span class="nb">len</span><span class="p">(</span><span class="n">o</span><span class="o">.</span><span class="n">idl</span><span class="p">[</span><span class="n">repname</span><span class="p">]):</span>
|
||||
</span><span id="create_dobs_string-800"><a href="#create_dobs_string-800"><span class="linenos">800</span></a> <span class="n">counters</span><span class="p">[</span><span class="n">oi</span><span class="p">]</span> <span class="o">=</span> <span class="o">-</span><span class="mi">1</span>
|
||||
</span><span id="create_dobs_string-801"><a href="#create_dobs_string-801"><span class="linenos">801</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
</span><span id="create_dobs_string-802"><a href="#create_dobs_string-802"><span class="linenos">802</span></a> <span class="n">num</span> <span class="o">=</span> <span class="n">offsets</span><span class="p">[</span><span class="n">oi</span><span class="p">]</span>
|
||||
</span><span id="create_dobs_string-802"><a href="#create_dobs_string-802"><span class="linenos">802</span></a> <span class="n">num</span> <span class="o">=</span> <span class="mi">0</span>
|
||||
</span><span id="create_dobs_string-803"><a href="#create_dobs_string-803"><span class="linenos">803</span></a> <span class="k">if</span> <span class="n">num</span> <span class="o">==</span> <span class="mi">0</span><span class="p">:</span>
|
||||
</span><span id="create_dobs_string-804"><a href="#create_dobs_string-804"><span class="linenos">804</span></a> <span class="n">data</span> <span class="o">+=</span> <span class="s1">'0 '</span>
|
||||
</span><span id="create_dobs_string-805"><a href="#create_dobs_string-805"><span class="linenos">805</span></a> <span class="k">else</span><span class="p">:</span>
|
||||
|
|
|
|||
|
|
@ -1304,7 +1304,7 @@
|
|||
</span><span id="L-1097"><a href="#L-1097"><span class="linenos">1097</span></a> <span class="n">ret</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">zeros</span><span class="p">(</span><span class="n">new_idx</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">new_idx</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span> <span class="o">+</span> <span class="mi">1</span><span class="p">)</span>
|
||||
</span><span id="L-1098"><a href="#L-1098"><span class="linenos">1098</span></a> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">shape</span><span class="p">):</span>
|
||||
</span><span id="L-1099"><a href="#L-1099"><span class="linenos">1099</span></a> <span class="n">ret</span><span class="p">[</span><span class="n">idx</span><span class="p">[</span><span class="n">i</span><span class="p">]</span> <span class="o">-</span> <span class="n">new_idx</span><span class="p">[</span><span class="mi">0</span><span class="p">]]</span> <span class="o">=</span> <span class="n">deltas</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
|
||||
</span><span id="L-1100"><a href="#L-1100"><span class="linenos">1100</span></a> <span class="k">return</span> <span class="n">np</span><span class="o">.</span><span class="n">array</span><span class="p">([</span><span class="n">ret</span><span class="p">[</span><span class="n">new_idx</span><span class="p">[</span><span class="n">i</span><span class="p">]</span> <span class="o">-</span> <span class="n">new_idx</span><span class="p">[</span><span class="mi">0</span><span class="p">]]</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">new_idx</span><span class="p">))])</span>
|
||||
</span><span id="L-1100"><a href="#L-1100"><span class="linenos">1100</span></a> <span class="k">return</span> <span class="n">np</span><span class="o">.</span><span class="n">array</span><span class="p">([</span><span class="n">ret</span><span class="p">[</span><span class="n">new_idx</span><span class="p">[</span><span class="n">i</span><span class="p">]</span> <span class="o">-</span> <span class="n">new_idx</span><span class="p">[</span><span class="mi">0</span><span class="p">]]</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">new_idx</span><span class="p">))])</span> <span class="o">*</span> <span class="nb">len</span><span class="p">(</span><span class="n">new_idx</span><span class="p">)</span> <span class="o">/</span> <span class="nb">len</span><span class="p">(</span><span class="n">idx</span><span class="p">)</span>
|
||||
</span><span id="L-1101"><a href="#L-1101"><span class="linenos">1101</span></a>
|
||||
</span><span id="L-1102"><a href="#L-1102"><span class="linenos">1102</span></a>
|
||||
</span><span id="L-1103"><a href="#L-1103"><span class="linenos">1103</span></a><span class="k">def</span> <span class="nf">derived_observable</span><span class="p">(</span><span class="n">func</span><span class="p">,</span> <span class="n">data</span><span class="p">,</span> <span class="n">array_mode</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span class="p">):</span>
|
||||
|
|
|
|||
File diff suppressed because one or more lines are too long
Loading…
Add table
Add a link
Reference in a new issue