Python - Cách chọn một cột từ Pandas DataFrame

Để chọn một cột từ DataFrame, chỉ cần tìm nạp nó bằng dấu ngoặc vuông. Đề cập đến cột để chọn trong dấu ngoặc và đó là nó, chẳng hạn như

dataFrame[‘ColumnName’]

Đầu tiên, hãy nhập thư viện được yêu cầu -

import pandas as pd

Bây giờ, hãy tạo một DataFrame. Chúng tôi có hai cột trong đó -

dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

Để chỉ chọn một cột duy nhất, hãy đề cập đến tên cột bằng cách sử dụng dấu ngoặc vuông như hình dưới đây. Ở đây, tên cột của chúng tôi là 'Xe hơi' -

dataFrame ['Car']

Ví dụ

Sau đây là mã -

import pandas as pd

# Create DataFrame
dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

print"DataFrame ...\n",dataFrame

# selecting a column
print"\nSelecting and displaying only a single column = \n",dataFrame ['Car']

Đầu ra

Điều này sẽ tạo ra kết quả sau -

DataFrame ...
       Car   Units
0      BMW     100
1    Lexus     150
2     Audi     110
3  Mustang      80
4  Bentley     110
5   Jaguar      90

Selecting and displaying only a single column =
0      BMW
1    Lexus
2     Audi
3  Mustang
4  Bentley
5   Jaguar
Name: Car, dtype: object