Make a list of the column headers in a Python Pandas DataFrame
David Y.
—The Problem
How can I create a list of the column headers in a Python Pandas DataFrame?
The Solution
The simplest way to do this is by directly casting the DataFrame to a list:
Click to Copy
import pandas data = { "A": [1, 2, 3], "B": [4, 5, 6], "C": [7, 8, 9] } df = pandas.DataFrame(data) column_names = list(df) print(column_names) # will print ['A', 'B', 'C']
A more explicit and performant method would be to use df.columns.values and its tolist method:
Click to Copy
import pandas data = { "A": [1, 2, 3], "B": [4, 5, 6], "C": [7, 8, 9] } df = pandas.DataFrame(data) column_names = df.columns.values.tolist() print(column_names) # will print ['A', 'B', 'C']
- Sentry BlogPython Performance Testing: A Comprehensive Guide (opens in a new tab)
- Syntax.fmListen to the Syntax Podcast (opens in a new tab)
- Sentry BlogLogging in Python: A Developer’s Guide (opens in a new tab)
- CodecovPython - Codecov (opens in a new tab)
- Listen to the Syntax Podcast (opens in a new tab)
![Syntax.fm logo]()
Tasty treats for web developers brought to you by Sentry. Get tips and tricks from Wes Bos and Scott Tolinski.
SEE EPISODES
Considered “not bad” by 4 million developers and more than 150,000 organizations worldwide, Sentry provides code-level observability to many of the world’s best-known companies like Disney, Peloton, Cloudflare, Eventbrite, Slack, Supercell, and Rockstar Games. Each month we process billions of exceptions from the most popular products on the internet.