Python gap_fill_by_3の例

コード例 #1

def write_csv_for_plot(results, outfile, metadata=None, column_name=None):
    '''
    Writes a csv file to the specified outfile file-like object.
    Tweak the data to fill in gaps
    '''

    csv_writer = csv.writer(outfile)
    if metadata != None:
        csv_writer.writerow(metadata)
    result_keys = results.keys()
    if column_name != None:
        result_keys.append(column_name)
    csv_writer.writerow(results.keys())
    # Iterate, because csv.writerows pulls up all rows to a list
    if column_name == None:
        for r in gap_fill_by_3(results):
            csv_writer.writerow(r)
    else:
        extra = (None, )
        for r in gap_fill_by_3(results):
            # have to convert to tuple because RowProxy does not support +
            csv_writer.writerow(tuple(r) + extra)

コード例 #2

ファイル: stage_data.py プロジェクト: ayan-usgs/EDEN-hydrograph-data

def write_csv_for_plot(results, outfile, metadata=None, column_name=None):
    '''
    Writes a csv file to the specified outfile file-like object.
    Tweak the data to fill in gaps
    '''

    csv_writer = csv.writer(outfile)
    if metadata != None:
        csv_writer.writerow(metadata)
    result_keys = results.keys()
    if column_name != None:
        result_keys.append(column_name)
    csv_writer.writerow(results.keys())
    # Iterate, because csv.writerows pulls up all rows to a list
    if column_name == None:
        for r in gap_fill_by_3(results):
            csv_writer.writerow(r)
    else:
        extra = (None,)
        for r in gap_fill_by_3(results):
            # have to convert to tuple because RowProxy does not support +
            csv_writer.writerow(tuple(r) + extra)

コード例 #3

ファイル: gap_fill_test.py プロジェクト: ayan-usgs/EDEN-hydrograph-data

 def test_base_case_by_3(self):
     ssz = [('date', 100, None, None, None, None, None),
           ('date', 101, None, None, 0, None, None),
           ('date', None, 102, None, 0, None, None),
           ('date', 103, None, None, None, None, None),
           ('date', None, None, 104, None, None, 1),
           ('date', None, None, 105, None, None, 2),
           ('date', None, None, 106, None, None, None),
           ]
     gfg = gap_fill_by_3(ssz)
     l = list(gfg)
     expected = [
           ('date', 100, None, None, None, None, None),
           ('date', 101, None, None, 0, None, None),
           ('date', 102, 102, None, 0, None, None),
           ('date', 103, 103, None, None, None, None),
           ('date', 104, None, 104, None, None, 1),
           ('date', None, None, 105, None, None, 2),
           ('date', None, None, 106, None, None, None),
           ]
     self.assertEquals(l, expected)

コード例 #4

ファイル: gap_fill_test.py プロジェクト: govtmirror/EDEN-hydrograph-data

 def test_base_case_by_3(self):
     ssz = [
         ('date', 100, None, None, None, None, None),
         ('date', 101, None, None, 0, None, None),
         ('date', None, 102, None, 0, None, None),
         ('date', 103, None, None, None, None, None),
         ('date', None, None, 104, None, None, 1),
         ('date', None, None, 105, None, None, 2),
         ('date', None, None, 106, None, None, None),
     ]
     gfg = gap_fill_by_3(ssz)
     l = list(gfg)
     expected = [
         ('date', 100, None, None, None, None, None),
         ('date', 101, None, None, 0, None, None),
         ('date', 102, 102, None, 0, None, None),
         ('date', 103, 103, None, None, None, None),
         ('date', 104, None, 104, None, None, 1),
         ('date', None, None, 105, None, None, 2),
         ('date', None, None, 106, None, None, None),
     ]
     self.assertEquals(l, expected)