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

Python Pandas - สร้างชุดย่อยโดยเลือกค่าเฉพาะจากคอลัมน์ตามดัชนี


หากต้องการสร้างชุดย่อยโดยเลือกค่าเฉพาะจากคอลัมน์ตามดัชนี ให้ใช้เมธอด iloc() ให้เรานำเข้าห้องสมุดแพนด้าก่อน

import pandas as pd

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

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

การสร้างเซตย่อยที่มี 2 คอลัมน์และ 2 แถวแรกโดยใช้ iloc(

print"\nDisplaying a subset using iloc() = \n",dataFrame.iloc[0:2, 0:2]

ตัวอย่าง

ต่อไปนี้เป็นรหัสที่สมบูรณ์

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

# creating a subset with 2 columns and 1st 2 rows using iloc()
print"\nDisplaying a subset using iloc() = \n",dataFrame.iloc[0:2, 0:2]

ผลลัพธ์

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

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

Displaying a subset using iloc() =
   Closing_Stock   Opening_Stock
0           200            300
1           500            700