Pandas tạo khung dữ liệu để xử lý dữ liệu trong chương trình python. Trong bài viết này, chúng ta sẽ xem cách chúng ta có thể thêm một cột mới vào khung dữ liệu hiện có dựa trên các điều kiện nhất định.
Khung dữ liệu đã cho
Dưới đây là DataFrame gấu trúc đã cho mà chúng tôi sẽ thêm các cột bổ sung. Nó mô tả Ngày và Chủ đề của một cuộc kiểm tra.
Ví dụ
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) Đầu ra
Chạy đoạn mã trên cho chúng ta kết quả sau -
Exam Day Exam Subject 0 Mon Chemisry 1 Tue Physics 2 Wed Maths 3 Thu English 4 Fri Biology
Thêm một cột mới
Tiếp theo, chúng tôi quyết định thêm một cột khác xác định thời gian của kỳ thi. Ở đây, chúng tôi thêm điều kiện bằng cách sử dụng câu lệnh if và đặt tên cột bổ sung là Thời gian.
Ví dụ
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) Đầu ra
Chạy đoạn mã trên cho chúng ta kết quả sau -
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