Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

Python - วิธีเลือกคอลัมน์จาก Pandas DataFrame


ในการเลือกคอลัมน์จาก DataFrame ให้ดึงข้อมูลโดยใช้วงเล็บเหลี่ยม ระบุคอลัมน์ที่จะเลือกในวงเล็บ แค่นั้น เช่น

dataFrame[‘ColumnName’]

ขั้นแรก นำเข้าไลบรารีที่จำเป็น -

import pandas as pd

ตอนนี้สร้าง DataFrame เรามีสองคอลัมน์ในนั้น -

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

หากต้องการเลือกเพียงคอลัมน์เดียว ให้ระบุชื่อคอลัมน์โดยใช้วงเล็บเหลี่ยมดังที่แสดงด้านล่าง ที่นี่ชื่อคอลัมน์ของเราคือ 'รถยนต์' −

dataFrame ['Car']

ตัวอย่าง

ต่อไปนี้เป็นรหัส -

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']

ผลลัพธ์

สิ่งนี้จะสร้างผลลัพธ์ต่อไปนี้ -

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