Extract csv file specific columns to list in Python

PythonMatplotlibServer Side ProgrammingProgramming

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 −

NameMarks
Arun98
Shyam75
Govind54
Javed92
Raju87

Output

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

raja
Published on 07-May-2021 07:45:01
Advertisements