Examples of how to iterate over the first n rows of a pandas DataFrame in python
Create a pandas dataframe
Let's first create a DataFrame:
import pandas as pd
import numpy as np
data = np.arange(1,31)
data = data.reshape(10,3)
df = pd.DataFrame(data, columns=['A','B','C'])
print(df)
gives
A B C
0 1 2 3
1 4 5 6
2 7 8 9
3 10 11 12
4 13 14 15
5 16 17 18
6 19 20 21
7 22 23 24
8 25 26 27
9 28 29 30
Iterate over the first n rows using head()
A first solution is to use the pandas head() method:
n = 4
df.head(n)
will display here the first 4 rows:
A B C
0 1 2 3
1 4 5 6
2 7 8 9
3 10 11 12
Using iloc()
A second approach is to use iloc()
df.iloc[:n,:]
gives
A B C
0 1 2 3
1 4 5 6
2 7 8 9
3 10 11 12
Using iterrows()
Another possible solution is to use iterrows()
for index,row in df.iterrows():
if index < n:
print(row)
print()
A 1
B 2
C 3
Name: 0, dtype: int64
A 4
B 5
C 6
Name: 1, dtype: int64
A 7
B 8
C 9
Name: 2, dtype: int64
A 10
B 11
C 12
Name: 3, dtype: int64