Cách lấy chỉ số cột từ tên cột trong Python Pandas

Để lấy chỉ số (index) cột từ tên cột trong Python Pandas, ta dùng phương thức get_loc().

Các bước

• Tạo dữ liệu bảng hai chiều df.
• In DataFrame đầu vào df.
• Lấy danh sách cột dùng df.columns.
• In danh sách cột.
• Khởi tạo biến column_name.
• Lấy vị trí (index) của column_name bằng get_loc().
• In chỉ số cột.

Ví dụ

import pandas as pd

df = pd.DataFrame({
    "x": [5, 2, 7, 0],
    "y": [4, 7, 5, 1],
    "z": [9, 3, 5, 1]
})

print("Input DataFrame 1 is:\n", df)

columns = df.columns
print("Các cột trong DataFrame:", columns)

column_name = "z"
column_index = columns.get_loc(column_name)
print("Chỉ số cột", column_name, "là:", column_index)

column_name = "x"
column_index = columns.get_loc(column_name)
print("Chỉ số cột", column_name, "là:", column_index)

column_name = "y"
column_index = columns.get_loc(column_name)
print("Chỉ số cột", column_name, "là:", column_index)

Kết quả đầu ra

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

Các cột trong DataFrame: Index(['x', 'y', 'z'], dtype='object')
Chỉ số cột z là: 2
Chỉ số cột x là: 0
Chỉ số cột y là: 1

Lưu ý: get_loc() trả về vị trí dựa trên 0 (zero-based). Nếu cột không tồn tại, sẽ ném ra KeyError.