How can I create a list of the column headers in a Python Pandas DataFrame?
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 actionable, code-level insights to resolve Python performance bottlenecks and errors.
Create a free Sentry account
Create a Python project and note your DSN
Grab the Sentry Python SDK
pip install --upgrade sentry-sdk
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, )
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.