DataFrame is a two-dimensional data structure, which consists of labeled rows and columns. The number of rows and columns give us the shape of the DataFrame, and therefore is an indication of the data size. In this short how-to article, we will learn how to find the row count of Pandas and PySpark DataFrames.
Pandas
There are two ways of getting the row count. The first one is the built-in len function of Python, which can be applied to different data structuctures. For instance, we can use the len function to find the number of items in a list. When applied to a DataFrame, it gives us the row count.
len(df)
10000
The other one is the shape method, which returns a tuple that contains both the number of rows and columns. We can get the row count by selecting the first item of the tuple returned by the shape method.
df.shape[0]
10000
PySpark
The count function gives us the row count.
df.count()
10000
This question is also being asked as:
- How to count the number of records or rows in a DataFrame