'Python script to do a vlookup in excel

I'm trying to write a Python script to do a vlookup from an Excel sheet. I want to add an option in the script that user can write and search for a specific value so the script provide the entire information for it.

script:

import pandas as pd

df_1 = pd.read_excel("Process_Class_Generator.xlsx",sheet_name="Data")
df_2 = pd.read_excel("Process_Class_Generator.xlsx",sheet_name="PC")

print(df_1.columns)
print(df_2.columns)


df_2.rename(columns={"Serial":"ID"},inplace=True)

df_3 = pd.merge(df_1,df_2[["Process Class","ID"]],on="ID",how="left")

print(df_3)

with pd.ExcelWriter("output.xlsx") as writer:
    df_2.to_excel(writer, sheet_name="Result")
    df_3.to_excel(writer,sheet_name="Final")


Sources

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Source: Stack Overflow

Solution Source