diff --git a/pydfs_lineup_optimizer/lineup_exporter.py b/pydfs_lineup_optimizer/lineup_exporter.py index 540afc1..4cf2484 100644 --- a/pydfs_lineup_optimizer/lineup_exporter.py +++ b/pydfs_lineup_optimizer/lineup_exporter.py @@ -46,6 +46,23 @@ def export(self, filename, render_func=None): lineup_writer.writerow(row) +class DataFrameLineupExporter(LineupExporter): + def export(self, render_func=None): + rows = [] + try: + for index, lineup in enumerate(self.lineups): + if index == 0: + pass + row = [(render_func or self.render_player)(player) for player in lineup.lineup] + row.append(str(lineup.salary_costs)) + row.append(str(lineup.fantasy_points_projection)) + rows.append(row) + import pandas as pd + return pd.DataFrame(rows) + except (ImportError, ModuleNotFoundError): + return rows + + class FantasyDraftCSVLineupExporter(LineupExporter): def export(self, filename, render_func=None): if not self.lineups: