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:

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:

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']

Get Started With Sentry

Get actionable, code-level insights to resolve Python performance bottlenecks and errors.

  1. Create a free Sentry account

  2. Create a Python project and note your DSN

  3. Grab the Sentry Python SDK

pip install --upgrade sentry-sdk
  1. Configure your DSN
import sentry_sdk sentry_sdk.init( "https://<key>@sentry.io/<project>", # Set traces_sample_rate to 1.0 to capture 100% # of transactions for performance monitoring. # We recommend adjusting this value in production. traces_sample_rate=1.0, )

Check our documentation for the latest instructions.

Loved by over 4 million developers and more than 90,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.

Share on Twitter
Bookmark this page
Ask a questionJoin the discussion

Related Answers

A better experience for your users. An easier life for your developers.

    TwitterGitHubDribbbleLinkedinDiscord
© 2024 • Sentry is a registered Trademark
of Functional Software, Inc.