열 추가하기

기본 코드

import pandas as pd
import numpy as np

friend_dict_list = [
    {'name': 'Jone', 'age': 15, 'job': 'student'},
    {'name': 'Jenny', 'age': 30, 'job': 'developer'},
    {'name': 'Nate', 'age': 30, 'job': 'teacher'}
]

df = pd.DataFrame(friend_dict_list, columns = ['name', 'age', 'job'])

• 새로운 열을 기본값과 함께 추가

df['salary'] = 0
df
    name  age        job  salary
0   Jone   15    student       0
1  Jenny   30  developer       0
2   Nate   30    teacher       0

• 기본 값이 아닌 다른 값 넣기 : numpy 이용해서 가능하다.

df['salary'] = np.where(df['job'] != 'student' , 'yes', 'no')
df
    name  age        job salary
0   Jone   15    student     no
1  Jenny   30  developer    yes
2   Nate   30    teacher    yes

기본 코드

import pandas as pd
import numpy as np

friend_dict_list = [
    {'name': 'John', 'midterm': 95, 'final': 85},
    {'name': 'Jenny', 'midterm': 85, 'final': 80},
    {'name': 'Nate', 'midterm': 10, 'final': 30}
]

df = pd.DataFrame(friend_dict_list, columns = ['name', 'midterm', 'final'])

name  midterm  final
0   John       95     85
1  Jenny       85     80
2   Nate       10     30

• 기존의 두 열 연산으로 새로운 열 만들기

df['total'] = df['midterm'] + df['final']
df
    name  midterm  final  total
0   John       95     85    180
1  Jenny       85     80    165
2   Nate       10     30     40

평균값 구하기

df['average'] = df['total'] / 2
df
    name  midterm  final  total  average
0   John       95     85    180     90.0
1  Jenny       85     80    165     82.5
2   Nate       10     30     40     20.0

• 리스트를 이용해 열 추가하기

grades = []

for row in df['average']:
    if row >= 90:
        grades.append('A')
    elif row >= 80:
        grades.append('B')
    elif row >= 70:
        grades.append('C')
    else:
        grades.append('F')
        
df['grade'] = grades

    name	midterm	final	total	average	grade
0	  John	  95	  85	  180	   90.0	    A
1	  Jenny	  85	  80	  165	   82.5	    B
2	   Nate	  10	  30	  40	   20.0	    F