// Databricks notebook source
val df = spark.read.json("/mnt/demo/small_radio_json.json")

// COMMAND ----------

df.show()

// COMMAND ----------

val specificColumnsDf = df.select("firstname", "lastname", "gender", "location", "level")
specificColumnsDf.show()

// COMMAND ----------

val renamedColumnsDF = specificColumnsDf.withColumnRenamed("level", "subscription_type")
renamedColumnsDF.show()


// COMMAND ----------

renamedColumnsDF.write.mode("overwrite").json("/mnt/demo/output/renamedColumnsDF.json")