docs: def edit_ipython_profile(spark_home, spark_python, py4j):
      """Adds a startup file to the current IPython profile to import pyspark.
  
      The startup file sets the required environment variables and imports pyspark.
  
      Parameters
      ----------
      spark_home : str
          Path to Spark installation.
      spark_python : str
          Path to python subdirectory of Spark installation.
      py4j : str
          Path to py4j library.
      """
      from IPython import get_ipython
      ip = get_ipython()
  
      if ip:
          profile_dir = ip.profile_dir.location
      else:
          from IPython.utils.path import locate_profile
          profile_dir = locate_profile()
  
      startup_file_loc = os.path.join(profile_dir, "startup", "findspark.py")
  
      with open(startup_file_loc, 'w') as startup_file:
          #Lines of code to be run when IPython starts
          startup_file.write("import sys, os\n")
          startup_file.write("os.environ['SPARK_HOME'] = '" + spark_home + "'\n")
          startup_file.write("sys.path[:0] = " + str([spark_python, py4j]) + "\n")
          startup_file.write("import pyspark\n")