Skip to content

Commit

Permalink
Fix the imports, yapf (#33)
Browse files Browse the repository at this point in the history
  • Loading branch information
Montana committed Jul 13, 2023
1 parent e25faae commit 4ec02f7
Showing 1 changed file with 11 additions and 13 deletions.
24 changes: 11 additions & 13 deletions metricsVisualizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,10 @@
import matplotlib.pyplot as plt
import pandas as pd

def plotData(csvFile, column, prefix, outputdir):
df = pd.read_csv(csvFile)
outputFile=os.path.join(outputdir, f"{prefix}_{column.lower()}.png")

def plot_data(csv_file, column, prefix, output_dir):
df = pd.read_csv(csv_file)
output_file = os.path.join(output_dir, f"{prefix}_{column.lower()}.png")

pd.options.display.float_format = '{:.2f}'.format

Expand All @@ -21,20 +22,17 @@ def plotData(csvFile, column, prefix, outputdir):
# Rotating X-axis labels
plt.xticks(rotation=25)

lineplot.figure.savefig(outputFile)
lineplot.figure.savefig(output_file)


if __name__ == "__main__":
parser = argparse.ArgumentParser(
description='Visualizes resource usage samples as graphs.')
parser.add_argument('-i', '--input',
help="The input CSV file.", required=True)
parser.add_argument('-c', '--column',
help="The column to sum.", required=True)
parser.add_argument('-p', '--prefix',
help="The output file's prefix.", required=True)
parser.add_argument('-o', '--outputdir',
help="The output directory.", required=True)
parser.add_argument('-i', '--input', help="The input CSV file.", required=True)
parser.add_argument('-c', '--column', help="The column to sum.", required=True)
parser.add_argument('-p', '--prefix', help="The output file's prefix.", required=True)
parser.add_argument('-o', '--outputdir', help="The output directory.", required=True)

args = parser.parse_args()

plotData(args.input, args.column, args.prefix, args.outputdir)
plot_data(args.input, args.column, args.prefix, args.outputdir)

0 comments on commit 4ec02f7

Please sign in to comment.