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

การสร้างคอลัมน์ Pandas dataframe ตามเงื่อนไขที่กำหนดใน Python


Pandas สร้าง data frames เพื่อประมวลผลข้อมูลในโปรแกรม python ในบทความนี้ เราจะมาดูกันว่าเราจะเพิ่มคอลัมน์ใหม่ให้กับ dataframe ที่มีอยู่ได้อย่างไรโดยอิงตามเงื่อนไขบางประการ

กรอบข้อมูลที่กำหนด

ด้านล่างนี้คือ DataFrame แพนด้าที่กำหนด ซึ่งเราจะเพิ่มคอลัมน์เพิ่มเติม มันอธิบายวันและวิชาของการสอบ

ตัวอย่าง

import pandas as pd

# Lists for Exam subjects and Days
Days = ['Mon', 'Tue', 'Wed','Thu', 'Fri']
Sub = ['Chemisry','Physics','Maths','English','Biology']

# Dictionary for Exam Schedule
Exam_Subjects = {'Exam Day': Days,
               'Exam Subject': Sub}

# Dictionary to DataFrame
Exam_Subjects_df = pd.DataFrame(Exam_Subjects)
print(Exam_Subjects_df)

ผลลัพธ์

การเรียกใช้โค้ดข้างต้นทำให้เราได้ผลลัพธ์ดังต่อไปนี้ -

   Exam Day  Exam Subject
0    Mon     Chemisry
1    Tue     Physics
2    Wed     Maths
3    Thu     English
4    Fri     Biology

เพิ่มคอลัมน์ใหม่

ต่อไปเราตัดสินใจเพิ่มคอลัมน์อื่นที่ระบุเวลาของการสอบ ที่นี่เราเพิ่มเงื่อนไขโดยใช้คำสั่ง if และตั้งชื่อคอลัมน์เพิ่มเติมเป็น Time

ตัวอย่าง

import pandas as pd

# Lists for Exam subjects
Days = ['Mon', 'Tue', 'Wed','Thu', 'Fri']
Sub = ['Chemisry','Physics','Maths','English','Biology']


# Dictionary for Exam Schedule
Exam_Subjects = {'Exam Day': Days,
                  'Exam Subject': Sub}

# Dictionary to DataFrame
Exam_Subjects_df = pd.DataFrame(Exam_Subjects)

Exam_Subjects_df['Time'] = ['2 PM' if x in('Mon','Thu') else '10 AM' for x in Exam_Subjects_df['Exam Day']]
print(Exam_Subjects_df)

ผลลัพธ์

การเรียกใช้โค้ดข้างต้นทำให้เราได้ผลลัพธ์ดังต่อไปนี้ -

  Exam Day   Exam Subject  Time
0  Mon       Chemisry      2 PM
1  Tue       Physics       10 AM
2  Wed       Maths         10 AM
3  Thu       English       2 PM
4  Fri       Biology       10 AM