2021. 03. 09 화요일

Intermediate Python

Loops

Loop over dictionary

• 사전에서 for 루프를 통해 키와 값을 불러오기 위해서는 items() 메소드를 사용해야한다.

for key, value in dict.items() :
	print(key, value)

Loop over numpy array

• 어레이에서 for 루프를 통해 모든 값을 각각 나열하기 위해서는 nditer() 함수를 사용해야 한다.

for value in numpy.nditer(array) :
	print(value)

Loop over DataFrame

• 데이터 프레임에서 for 일반적인 루프를 사용하면 Column명만 나열된다.
  따라서 데이터 프레임에 포함된 데이터를 모두 불러오기 위해서는 iterrows() 메소드를 사용해야한다.

import pandas as pd
df = pd.read_csv('df.csv', index_col = 0)

for lab, row in df.iterrows() :
	print(lab)
    print(row)

• 특정 Column만 불러오기

import pandas as pd
df = pd.read_csv('df.csv', index_col = 0)

for lab, row in df.iterrows() :
	print(lab)
    print(row['column'])

• column 추가하기 (loop 사용)

import pandas as pd
df = pd.read_csv('df.csv', index_col = 0)

for lab, row in df.iterrows() :
	df.loc(lab, '추가할 column명') = 추가할 값
    
# 예시 :  df.loc(lab, 'column_length') = len(row['column'])

• column 추가하기 (loop 미사용)

import pandas as pd
df = pd.read_csv('df.csv', index_col = 0)

df['add_column'] = df['column'].apply(len)