David Y.
—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']
Tasty treats for web developers brought to you by Sentry. Get tips and tricks from Wes Bos and Scott Tolinski.
SEE EPISODESConsidered “not bad” by 4 million developers and more than 100,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.