Reading CSV files with options in Apache Spark

Introduction

We use options to tell Spark exactly how to read a CSV file. This helps Spark understand the file better and get the data right.

When the CSV file has a header row with column names.

When the data uses a separator other than a comma, like a semicolon or tab.

When you want to ignore bad or corrupted lines in the file.

When you want to specify how to handle missing values.

When you want to control how Spark infers data types.

Syntax

Apache Spark

spark.read.option("option_name", "option_value").csv("file_path")

You can chain multiple option() calls to set different options.

Common options include header, sep, inferSchema, and mode.

Examples

Reads a CSV file where the first row has column names.

Apache Spark

df = spark.read.option("header", "true").csv("data.csv")

Reads a CSV file with semicolon separators and a header row.

Apache Spark

df = spark.read.option("sep", ";").option("header", "true").csv("data.csv")

Reads a CSV file and tries to guess the data types of each column.

Apache Spark

df = spark.read.option("inferSchema", "true").csv("data.csv")

Reads a CSV file and skips lines that are corrupted or badly formatted.

Apache Spark

df = spark.read.option("mode", "DROPMALFORMED").csv("data.csv")

Sample Program

This program reads a CSV file named example.csv that uses semicolons to separate values and has a header row. It also asks Spark to guess the data types of each column. Then it prints the schema and the data.

Apache Spark

from pyspark.sql import SparkSession

spark = SparkSession.builder.appName("CSVOptionsExample").getOrCreate()

# Read CSV with header, semicolon separator, and infer schema
file_path = "example.csv"
df = spark.read.option("header", "true")\
               .option("sep", ";")\
               .option("inferSchema", "true")\
               .csv(file_path)

# Show the data
print("DataFrame schema:")
df.printSchema()
print("DataFrame content:")
df.show()

spark.stop()

Important Notes

Setting header to true tells Spark to use the first row as column names.

The inferSchema option can slow down reading for big files but gives correct data types.

The mode option controls how Spark handles bad lines: PERMISSIVE (default), DROPMALFORMED, or FAILFAST.

Summary

Use options to customize how Spark reads CSV files.

Common options include header, sep, inferSchema, and mode.

Setting options correctly helps Spark read data accurately and avoid errors.

Practice

1. What does the header option do when reading a CSV file in Apache Spark?

easy

A. It tells Spark to treat the first line as column names.

B. It specifies the character used to separate columns.

C. It infers the data types of each column automatically.

D. It sets the file reading mode to permissive.

Solution

1. Step 1: Understand the header option

   The header option tells Spark whether the first line of the CSV file contains column names or data.

2. Step 2: Identify its effect

   If header is set to true, Spark uses the first line as column names; otherwise, it treats all lines as data.

3. Final Answer:

   It tells Spark to treat the first line as column names. -> Option A

4. Quick Check:

   header = first line as columns [OK]

Hint: Remember: header=true means first line is column names [OK]

Common Mistakes:

• Confusing header with separator option
• Thinking header infers data types
• Assuming header controls error handling

2. Which of the following is the correct way to read a CSV file with a semicolon separator and header in Spark?

easy

A. spark.read.csv(options={'sep': ';', 'header': 'true'}, 'file.csv')

B. spark.read.csv(sep=';', 'file.csv', header=True)

C. spark.read.option('sep', ';').option('header', 'true').csv('file.csv')

D. spark.read.csv(header=True, sep=';', 'file.csv')

Solution

1. Step 1: Recall Spark CSV reading syntax

   Spark uses option() method to set options before calling csv().

2. Step 2: Check each option usage

   spark.read.option('sep', ';').option('header', 'true').csv('file.csv') correctly chains option('sep', ';') and option('header', 'true') before reading the file.

3. Final Answer:

   spark.read.option('sep', ';').option('header', 'true').csv('file.csv') -> Option C

4. Quick Check:

   Use option() chaining for CSV options [OK]

Hint: Use .option() chaining to set CSV read options [OK]

Common Mistakes:

• Passing options directly inside csv() as kwargs
• Using Python booleans instead of strings for options
• Incorrect method chaining order

3. Given the code below, what will be the schema of the DataFrame?

df = spark.read.option('header', 'true').option('inferSchema', 'true').csv('data.csv')

Assuming data.csv has columns: id (integer), name (string), age (integer).

medium

A. All columns will be read as strings.

B. id and age as integers, name as string.

C. All columns will be inferred as integers.

D. Schema will be empty because no schema is provided.

Solution

1. Step 1: Understand inferSchema option

   When inferSchema is true, Spark tries to detect the data type of each column automatically.

2. Step 2: Apply to given columns

   Since id and age contain integers, they will be inferred as integer types; name remains string.

3. Final Answer:

   id and age as integers, name as string. -> Option B

4. Quick Check:

   inferSchema=true means detect types [OK]

Hint: inferSchema=true detects column data types automatically [OK]

Common Mistakes:

• Assuming all columns are strings by default
• Thinking inferSchema converts all to integers
• Believing schema is empty without explicit definition

4. What is wrong with the following code snippet for reading a CSV file with a header and comma separator?

df = spark.read.csv(header=True, sep=',', 'file.csv')

medium

A. The header option must be set to 'false' to read headers.

B. The separator should be a semicolon, not a comma.

C. The file path must be an absolute path.

D. The options should be set using the option() method, not as parameters in csv().

Solution

1. Step 1: Check how options are passed in Spark CSV reading

   Spark requires options like header and sep to be set using option() method before calling csv().

2. Step 2: Identify the error in the code

   The code incorrectly passes options as parameters inside csv(), which is not supported and causes an error.

3. Final Answer:

   The options should be set using the option() method, not as parameters in csv(). -> Option D

4. Quick Check:

   Use .option() for CSV options, not csv() params [OK]

Hint: Set CSV options with .option(), not inside csv() call [OK]

Common Mistakes:

• Passing options directly inside csv()
• Using Python booleans instead of strings for options
• Confusing separator character

5. You want to read a CSV file that may have some corrupted rows. Which option should you use to skip corrupted rows and still read the rest of the data?

hard

A. mode='dropmalformed'

B. mode='failfast'

C. mode='permissive'

D. mode='ignore'

Solution

1. Step 1: Understand CSV reading modes in Spark

   Spark supports modes like 'permissive', 'failfast', and 'dropmalformed' to handle corrupted rows.

2. Step 2: Identify mode to skip corrupted rows

   The 'dropmalformed' mode skips corrupted rows and reads the rest without error.

3. Final Answer:

   mode='dropmalformed' -> Option A

4. Quick Check:

   Use dropmalformed to skip bad rows [OK]

Hint: Use mode='dropmalformed' to skip bad rows safely [OK]

Common Mistakes:

• Using failfast which stops on error
• Assuming permissive skips corrupted rows
• Using ignore which is not a valid mode