Sentry Answers>Python>

Make a list of the column headers in a Python Pandas DataFrame

Make a list of the column headers in a Python Pandas DataFrame

David Y.

—

The ProblemJump To Solution

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
  • Sentry BlogLogging in Python: A Developer’s Guide
  • Syntax.fm logo
    Listen to the Syntax Podcast

    Tasty Treats for Web Developers brought to you by Sentry. Web development tips and tricks hosted by Wes Bos and Scott Tolinski

    Listen to Syntax

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.

learn more about sentryjoin our discord
© 2024 • Sentry is a registered Trademark
of Functional Software, Inc.