Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Python에서 주어진 조건을 기반으로 Pandas 데이터 프레임 열 만들기

<시간/>

Pandas는 파이썬 프로그램에서 데이터를 처리하기 위해 데이터 프레임을 생성합니다. 이 기사에서는 특정 조건에 따라 기존 데이터 프레임에 새 열을 추가하는 방법을 살펴보겠습니다.

주어진 데이터 프레임

아래는 추가 열을 추가할 주어진 pandas 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 문을 사용하여 조건을 추가하고 추가 열의 이름을 시간으로 지정합니다.

예시

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