How can I get column names from a DataFrame in Python?

To retrieve column names from a Pandas DataFrame in Python, you can use the .columns attribute. This attribute returns an Index object containing the column labels. Here's how you can do it:

1. Import Pandas:

- First, make sure you have the Pandas library installed and imported into your Python script.

2. Create or Load a DataFrame:

- You'll need a DataFrame to work with. You can create one or load it from a file.

3. Access the Column Names:

- Use the .columns attribute to get the column names.

4. Example Code:

import pandas as pd

# Sample DataFrame
data = {'Name': ['Alice', 'Bob', 'Charlie'],
'Age': [25, 30, 28],
'City': ['New York', 'London', 'Paris']}
df = pd.DataFrame(data)

# Get column names
column_names = df.columns

# Print column names
print(column_names)

# Convert to a list if needed
column_list = df.columns.tolist()
print(column_list)

5. Explanation:

- The df.columns returns a Pandas Index object, which is similar to a list but has additional functionalities. If you need a standard Python list, you can convert it using .tolist().

6. Output:

- The output of the above code will be:

Index(['Name', 'Age', 'City'], dtype='object')
['Name', 'Age', 'City']

By using the .columns attribute, you can easily access and manipulate the column names of your DataFrame for further analysis or processing.

Learn more about this