From a087a08dea9d1de1c62a6968c71cd46d3cdab924 Mon Sep 17 00:00:00 2001 From: Eric Truett Date: Fri, 29 Jan 2021 09:47:31 -0600 Subject: [PATCH] Adds DataFrameLineupExporter class to lineup_exporter.py --- pydfs_lineup_optimizer/lineup_exporter.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/pydfs_lineup_optimizer/lineup_exporter.py b/pydfs_lineup_optimizer/lineup_exporter.py index 1badb5a..db9a9d3 100644 --- a/pydfs_lineup_optimizer/lineup_exporter.py +++ b/pydfs_lineup_optimizer/lineup_exporter.py @@ -37,6 +37,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: