def as_dataframe(self, model_group_ids):
"""Return model-group-id subset of table as dataframe
Args:
model_group_ids (list) the desired model group ids
Returns: (pandas.DataFrame) The data from the table corresponding
to those model group ids
"""
return pd.read_sql(
'select * from {} where model_group_id in ({})'.format(
self.distance_table,
str_in_sql(model_group_ids)
),
self.db_engine
)
评论列表
文章目录