Comment itérer seulement sur les n premières lignes d'une dataframe pandas en python ?

Exemples de comment itérer sur les n premières lignes d'une dataframe pandas en python:

Créer une pandas dataframe

Commençons par créer une dataframe avec plusieurs lignes:

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)

donne

    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

Itérer sur les n premières lignes en utilisant head()

Une première solution est d'utiliser la méthode pandas head():

n = 4

df.head(n)

va afficher ici les 4 premières lignes:

    A   B   C
0   1   2   3
1   4   5   6
2   7   8   9
3  10  11  12

En utilisant iloc()

Une seconde approche est d'utiliser iloc()

df.iloc[:n,:]

donne

    A   B   C
0   1   2   3
1   4   5   6
2   7   8   9
3  10  11  12

En utilisant iterrows()

Une autre solution possible est de passer par 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

Voir aussi