Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Python - Cách chọn nhiều hàng từ DataFrame Pandas

Để chọn nhiều hàng từ DataFrame, dùng toán tử cắt slice : với chỉ số vị trí.

Ví dụ

import pandas as pd

# Tạo DataFrame với index tùy chỉnh
dataFrame = pd.DataFrame(
    [[10, 15], [20, 25], [30, 35], [40, 45]],
    index=['w', 'x', 'y', 'z'],
    columns=['a', 'b']
)

print("DataFrame...\n", dataFrame)

# Chọn 1 hàng bằng label (loc)
print("\nChọn hàng label 'z' (loc):")
print(dataFrame.loc['z'])

# Chọn 1 hàng bằng vị trí (iloc)
print("\nChọn hàng vị trí 1 (iloc):")
print(dataFrame.iloc[1])

# Chọn nhiều hàng bằng slice vị trí
print("\nChọn hàng 0 đến 2 (slice 0:2):")
print(dataFrame[0:2])

Kết quả đầu ra

DataFrame...
    a   b
w  10  15
x  20  25
y  30  35
z  40  45

Chọn hàng label 'z' (loc):
a    40
b    45
Name: z, dtype: int64

Chọn hàng vị trí 1 (iloc):
a    20
b    25
Name: x, dtype: int64

Chọn hàng 0 đến 2 (slice 0:2):
    a   b
w  10  15
x  20  25

Tóm tắt chọn hàng

  • df.loc['label'] → chọn theo nhãn index (trả về Series).
  • df.iloc[0] → chọn theo vị trí số (trả về Series).
  • df[0:2] → slice vị trí (trả về DataFrame).
  • df.loc['w':'y'] → slice nhãn (bao gồm cả 'y').
  • df.iloc[0:2] → slice vị trí (không bao gồm 2).