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

Python Pandas - เซ็ตย่อย DataFrame ตามชื่อคอลัมน์


หากต้องการสร้างชุดย่อยของ DataFrame ตามชื่อคอลัมน์ ให้ใช้วงเล็บเหลี่ยม ใช้ DataFrame กับวงเล็บเหลี่ยม (ตัวดำเนินการสร้างดัชนี) และชื่อคอลัมน์เฉพาะเช่นนี้ −

dataFrame[‘column_name’]

ขั้นแรก นำเข้าไลบรารีที่จำเป็นด้วยนามแฝง −

import pandas as pd

สร้าง Pandas DataFrame พร้อมบันทึกผลิตภัณฑ์ -

dataFrame = pd.DataFrame({"Product": ["SmartTV", "ChromeCast", "Speaker", "Earphone"],
"Opening_Stock": [300, 700, 1200, 1500], "Closing_Stock": [200, 500, 1000, 900]})

ให้เราดึงชุดย่อย เช่น เรากำลังดึงเฉพาะระเบียนคอลัมน์ผลิตภัณฑ์

dataFrame['Product']

ตัวอย่าง

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

import pandas as pd

dataFrame = pd.DataFrame({"Product": ["SmartTV", "ChromeCast", "Speaker", "Earphone"],"Opening_Stock": [300, 700, 1200, 1500],"Closing_Stock": [200, 500, 1000, 900]})

print"DataFrame...\n",dataFrame

print"\nDisplaying a subset:\n",dataFrame['Product']

ผลลัพธ์

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

DataFrame...
   Closing_Stock   Opening_Stock   Product
0       200            300         SmartTV
1       500            700         ChromeCast
2       1000           1200        Speaker
3       900            1500        Earphone

Displaying a subset:
0      SmartTV
1      ChromeCast
2      Speaker
3      Earphone
Name: Product, dtype: object