Computer >> Máy Tính >  >> Lập trình >> Python

Làm cách nào để lấy chỉ mục cột từ tên cột trong Python Pandas?

Để lấy chỉ mục cột từ tên cột trong Python Pandas, chúng ta có thể sử dụng get_loc () phương pháp.

Các bước -

  • Tạo dữ liệu dạng bảng hai chiều, có thể thay đổi kích thước, có khả năng không đồng nhất, df .
  • In DataFrame đầu vào, df .
  • Tìm các cột của DataFrame, sử dụng df.columns .
  • In các cột từ Bước 3.
  • Khởi tạo một biến column_name .
  • Nhận vị trí, tức là chỉ mục cho tên_mạch .
  • In chỉ mục của tên_mạch .

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"Columns in the given DataFrame: ", columns

column_name = "z"
column_index = columns.get_loc(column_name)
print"Index of the column ", column_name, " is: ", column_index

column_name = "x"
column_index = columns.get_loc(column_name)
print"Index of the column ", column_name, " is: ", column_index

column_name = "y"
column_index = columns.get_loc(column_name)
print"Index of the column ", column_name, " is: ", column_index

Đầ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

Columns in the given DataFrame: Index(['x', 'y', 'z'],
dtype='object')

Index of the column z is: 2
Index of the column x is: 0
Index of the column y is: 1