Get column index from column name of a given Pandas DataFrame
import pandas as pd # Sample DataFrame data = {‘A’: [1, 2, 3], ‘B’: [4, 5, 6], ‘C’: [7, 8, 9]} df = pd.DataFrame(data) # Function to get column index by column name def get_column_index(df, column_name): try: return df.columns.get_loc(column_name) except KeyError: return f”Column ‘{column_name}’ not found” # Example usage column_name = ‘B’ index = get_column_index(df, …
Get column index from column name of a given Pandas DataFrame Read More »