Cách truy xuất nhóm hàng trong Pandas DataFrame

Để truy xuất một nhóm hàng trong Pandas DataFrame, chúng ta có thể sử dụng phương thức loc(). Ví dụ, nếu chúng ta sử dụng df.loc[2:5], thì nó sẽ chọn tất cả các hàng từ 2 đến 5.

Các bước

• Tạo dữ liệu bảng hai chiều, df.
• In DataFrame đầu vào, df.
• Sử dụng df.loc[2:5] để chọn hàng từ 2 đến 5.
• In DataFrame.

Ví dụ

import pandas as pd
df = pd.DataFrame({
    "x": [5, 2, 7, 0, 7, 0, 5, 2],
    "y": [4, 7, 5, 1, 5, 1, 4, 7],
    "z": [9, 3, 5, 1, 5, 1, 9, 3]
})
print("Input DataFrame is:\n", df)
df = df.loc[2:5]
print("New DataFrame:\n", df)

Kết quả

Input DataFrame is:
   x  y  z
0  5  4  9
1  2  7  3
2  7  5  5
3  0  1  1
4  7  5  5
5  0  1  1
6  5  4  9
7  2  7  3
New DataFrame:
   x  y  z
2  7  5  5
3  0  1  1
4  7  5  5
5  0  1  1