본문 바로가기

Pandas

DataFrame의 행, 열 선택


import pandas as pd

exam_data = {'수학' : [90, 80, 70],
             '영어' : [98, 89, 95],
             '음악' : [85, 95, 100],
             '체육' : [100, 90, 90]}

df = pd.DataFrame(exam_data, index=['서준', '우현', '인아'])
print(df)
print("1============")

# 행 인덱스를 사용하여 행 1개 선택
label1 = df.loc['서준']
print(label1)
print("2============")

position1 = df.iloc[0]
print(position1)
print("3============")

# 행 인덱스를 사용하여 2개 이상의 행 선택
label2 = df.loc[['서준', '우현']]
print(label2)
print("4============")

position2 = df.iloc[[0,1]]
print(position2)
print("5============")

# 행 인덱스의 범위를 지정하여 행 선택
label3 = df.loc['서준':'우현']
print(label3)
print("6============")


position3 = df.iloc[0:1]
print(position3)
print("7============")


exam_data = {'이름' : ['서준', '우현', '인아'],
             '수학' : [90, 80, 70],
             '영어' : [98, 89, 95],
             '음악' : [85, 95, 100],
             '체육' : [100, 90, 90]}

df = pd.DataFrame(exam_data)
print(df)

print(type(df))

# 수학 점수 데이터만 선택, 변수 math1 저장
math1 = df['수학']
print(math1)
print(type(math1))
print("================")

# 영어 점수 데이터만 선택, 변수 english에 저장
english = df.영어
print(english)
print(type(english))

# 음악, 체육 점수 데이터를 선택 변수 music_gym 에 저장
music_gym = df[['음악', '체육']]
print(music_gym)
print(type(music_gym))
print("================")

# 수학 점수 데이타만 선택. 변수 math2에 저장
math2 = df[['수학']]
print(math2)
print(type(math2))

 


Where there is a will there is a way