[Python] Pandas - Dataframe 함수 모음

 

merge

result_df = pd.merge(df1, df2, how='left', on='key')

result_df = pd.merge(df1, df2, how='left', left_on='name1', right_on='name2')

 

drop column

df.drop(columns=['datetime'], inplace=True)

 

reset index

df.reset_index(inplace=True, drop=True)

 

new col with apply

df['new_col'] = df.apply(lambda x: json.loads(x['loc'])['c'][1], axis=1)

 

sort

# 열 이름 순서 axis=1

# 내림차순 ascending=False

df.sort_values(by=['datetime'], inplace=True, ascending=True)

 

rename col

df = df.rename(columns={'name' : 'new_name'})

 

string replace

df['name'] = df['name'].str.replace('a', 'b')

 

to_sql 이용한 Bulk insert

• [출처] https://tzara.tistory.com/119
• if_exists
  - fail : 테이블 있을 경우 오류, 없을 경우 테이블 새로 생성되고 insert
  - replace : 테이블 drop, 재 생성 후 insert
  - append : 기존 테이블에 insert

def pg_connect():

    url = 'postgresql://postgres:pw@localhost:5432/postgres'

    engine = create_engine(url, client_encoding='utf8', use_batch_mode=True)

    return engine

 

pg_engine = pg_connect()

df.to_sql('target_table', con=pg_engine, if_exists='append', chunksize=1000, index=False)

 

첫번째 행을 columns 으로

df.rename(columns=df.iloc[0]) df.drop(df.index[0])

ㅔㅛ소