######################################################################
# Miscellaneous examples for data manipulation
# 12-01-2021
######################################################################

import pandas as pd 

######################################################################
# demonstrate function pandas.Series.repeat()
# repeat row multiple times 
# variable 'count' enumerates numbers from 'start' to 'end'
######################################################################

ex = pd.DataFrame({'id': ['a', 'b', 'c', 'd'],
                   'start': [1, 3, 5, 7],
                   'end': [8, 3, 6, 9]})
print(ex)
rangelist = [range(i, j + 1) for (i, j) in zip(ex.start, ex.end)]
ex = ex.loc[ex.index.repeat(ex.end - ex.start + 1)]
ex['count'] = [item for sublist in rangelist for item in sublist]
print(ex) 

######################################################################
# THE END
######################################################################