Extract Specific Columns from CSV File to List in Python

To extract csv file for specific columns to list in Python, we can use Pandas read_csv() method.

Steps

• Make a list of columns that have to be extracted.

• Use read_csv() method to extract the csv file into data frame.

• Print the exracted data.

• Plot the data frame using plot() method.

• To display the figure, use show() method.

Example

import pandas as pd
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
columns = ["Name", "Marks"]
df = pd.read_csv("input.csv", usecols=columns)
print("Contents in csv file:\n", df)
plt.plot(df.Name, df.Marks)
plt.show()

The csv file contains the following data −

Output

When we execute the code, it will extract the data from the csv file and show the following plot −