How to get column and row names in DataFrame?
Last Updated :
22 Feb, 2022
While analyzing the real datasets which are often very huge in size, we might need to get the rows or index names and columns names in order to perform certain operations.
Note: For downloading the nba dataset used in the below examples Click Here
Getting row names in Pandas dataframe
First, let's create a simple dataframe with nba.csv
Python3
# Import pandas package
import pandas as pd
# making data frame
data = pd.read_csv("nba.csv")
# calling head() method
# storing in new variable
data_top = data.head(10)
# display
data_top
Output:
Now let's try to get the row name from the above dataset.
Method #1: Simply iterate over indices
Python3
# Import pandas package
import pandas as pd
# making data frame
data = pd.read_csv("nba.csv")
# calling head() method
# storing in new variable
data_top = data.head()
# iterating the columns
for row in data_top.index:
print(row, end = " ")
Output:
0 1 2 3 4 5 6 7 8 9
Method #2: Using rows with dataframe object
Python3
# Import pandas package
import pandas as pd
# making data frame
data = pd.read_csv("nba.csv")
# calling head() method
# storing in new variable
data_top = data.head()
# list(data_top) or
list(data_top.index)
Output:
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
Method #3: index.values method returns an array of indexes.
Python3
# Import pandas package
import pandas as pd
# making data frame
data = pd.read_csv("nba.csv")
# calling head() method
# storing in new variable
data_top = data.head()
list(data_top.index.values)
Output:
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
Method #4: Using the tolist() method with values given the list of indexes.
Python3
# Import pandas package
import pandas as pd
# making data frame
data = pd.read_csv("nba.csv")
# calling head() method
# storing in new variable
data_top = data.head()
list(data_top.index.values.tolist())
Output:
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
Method #5: Count the number of rows in dataframe
Since we have loaded only 10 top rows of the dataframe using the head() method, let's verify the total number of rows first.
Python3
# iterate the indices and print each one
for row in data.index:
print(row, end = " ")
Output:
Now, let's print the total count of the index.
Python3
# Import pandas package
import pandas as pd
# making data frame
data = pd.read_csv("nba.csv")
row_count = 0
# iterating over indices
for col in data.index:
row_count += 1
# print the row count
print(row_count)
Output:
458
Getting column names in Pandas dataframe
Now let's try to get the columns name from the nba.csv dataset.
Method #1: Simply iterating over columns
Python3
# Import pandas package
import pandas as pd
# making data frame
data = pd.read_csv("nba.csv")
# iterating the columns
for col in data.columns:
print(col)
Output:
Method #2: Using columns with dataframe object
Python3
# Import pandas package
import pandas as pd
# making data frame
data = pd.read_csv("nba.csv")
# list(data) or
list(data.columns)
Output:
Method #3: column.values method returns an array of indexes.
Python3
# Import pandas package
import pandas as pd
# making data frame
data = pd.read_csv("nba.csv")
list(data.columns.values)
Output:
Method #4: Using tolist() method with values with given the list of columns.
Python3
# Import pandas package
import pandas as pd
# making data frame
data = pd.read_csv("nba.csv")
list(data.columns.values.tolist())
Output:
Method #5: Using sorted() method
Sorted() method will return the list of columns sorted in alphabetical order.
Python3
# Import pandas package
import pandas as pd
# making data frame
data = pd.read_csv("nba.csv")
# using sorted() method
sorted(data)
Output:
Similar Reads
How to Get Column Names in Pandas Dataframe
While analyzing the real datasets which are often very huge in size, we might need to get the pandas column names in order to perform certain operations. The simplest way to get column names in Pandas is by using the .columns attribute of a DataFrame. Let's understand with a quick example:Pythonimpo
4 min read
How to get name of dataframe column in PySpark ?
In this article, we will discuss how to get the name of the Dataframe column in PySpark. To get the name of the columns present in the Dataframe we are using the columns function through this function we will get the list of all the column names present in the Dataframe. Syntax: df.columns We can a
3 min read
How to change dataframe column names in PySpark ?
In this article, we are going to see how to change the column names in the pyspark data frame. Let's create a Dataframe for demonstration: Python3 # Importing necessary libraries from pyspark.sql import SparkSession # Create a spark session spark = SparkSession.builder.appName('pyspark - example jo
3 min read
Add column names to dataframe in Pandas
Sometimes, Pandas DataFrames are created without column names, or with generic default names (like 0, 1, 2, etc.). Let's learn how to add column names to DataFrames in Pandas. Adding Column Names Directly to columns Attribute The simplest way to add column names is by directly assigning a list of co
3 min read
Pandas Append Rows & Columns to Empty DataFrame
Appending rows and columns to an empty DataFrame in pandas is useful when you want to incrementally add data to a table without predefining its structure. To immediately grasp the concept, here’s a quick example of appending rows and columns to an empty DataFrame using the concat() method, which is
4 min read
How to rename columns in Pandas DataFrame
In this article, we will see how to rename column in Pandas DataFrame. The simplest way to rename columns in a Pandas DataFrame is to use the rename() function. This method allows renaming specific columns by passing a dictionary, where keys are the old column names and values are the new column nam
4 min read
How to Iterate over rows and columns in PySpark dataframe
In this article, we will discuss how to iterate rows and columns in PySpark dataframe. Create the dataframe for demonstration: Python3 # importing module import pyspark # importing sparksession from pyspark.sql module from pyspark.sql import SparkSession # creating sparksession and giving an app nam
6 min read
How to Get First Column of Pandas DataFrame?
In this article, we will discuss how to get the first column of the pandas dataframe in Python programming language. Method 1: Using iloc[] function This function is used to get the first column using slice operator. for the rows we extract all of them, for columns specify the index for first column
4 min read
How to Show All Columns of a Pandas DataFrame?
Pandas limit the display of rows and columns, making it difficult to view the full data, so let's learn how to show all the columns of Pandas DataFrame. Using pd.set_option to Show All Pandas ColumnsPandas provides a set_option() function that allows you to configure various display options, includi
2 min read
How to name aggregate columns in PySpark DataFrame ?
In this article, we are going to see how to name aggregate columns in the Pyspark dataframe. We can do this by using alias after groupBy(). groupBy() is used to join two columns and it is used to aggregate the columns, alias is used to change the name of the new column which is formed by grouping da
2 min read