Update stockreturns.py

shalabh147 · Apr 8, 2021 · 504238e · 504238e
1 parent 9d3410f
commit 504238e
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/Lab7-Spark/stockreturns.py b/Lab7-Spark/stockreturns.py
@@ -1,7 +1,6 @@
 from pyspark.sql import SQLContext
 from pyspark import SparkContext
-# other required imports here
-
+
 if __name__ == "__main__":
     # create Spark context with necessary configuration
     spark = SparkContext("local", "Stock Returns")
@@ -10,9 +9,9 @@
     df = SQLContext(spark).read.option("header", True).csv("stock_prices.csv")
 
     # calculate daily percentage returns
-    df = df.withColumn("return", 100*(df["close"]-df["open"])/df["open"])
+    df = df.withColumn("return", ((df["close"]-df["open"])/df["open"])*100)
 
-    # average on company name
+    # average on date
     df = df.groupBy("date").avg().alias("avg_return")
 
     # save the average returns to output