파이썬 기초 - 디렉터리와 파일 입출력

디렉터리 관련 함수

import os

print("현재 디렉터리 : ", os.getcwd())    # 현재 파이썬 소스파일이 위치한 디렉터리(폴더)

os.chdir("C:/")

print("이동한 디렉터리 : ", os.getcwd()) 

os.listdir(os.getcwd())                  # C드라이브내의 모든 파일 목록 출력

print("폴더의 존재여부 확인 ", os.path.exists("C:/Program"))   # True 혹은 False로 출력

os.mkdir("C:/newfile_folder")                 # 새로운 디렉터리(폴더) 생성 (하나만 생성)

os.makedirs("C:/first_folder/second_folder")  # first_folder와 second_folder 두개의 폴더 생성

os.rmdir("C:/newfile_folder")                 # 하나의 폴더 삭제

os.removedirs("C:/first_folder/second_folder") # 경로내에 표시된 모든 폴더 삭제

 

텍스트 파일 다루기

file = open("C:/test.txt", "w")
file.write("텍스트파일에 데이터 덮어쓰기")
file.close()

file1 = open("C:/test.txt", "a")
file.write("텍스트파일에 이어서 쓰기")
file.close()

file2 = open("C:/test.txt", "r")  # 파일읽기
file2.readlines()
print(file2[0])    # \n기준으로 인덱스

 

 

xls 파일과 csv파일 열기 (openpyxl, csv, pandas 라이브러리 설치 필요)

import pandas as pd
import openpyxl, csv

grade = ['A+', 'A0', 'B+', 'B0', 'C+', 'C0', 'D+']
name = ['김길동', '박혹식', '홍성임', '방형식', '정지손', '성지건', '이백숙']

grade_matrix = pd.DataFrame()
grade_matrix['학점'] = grade
grade_matrix['이름'] = name
print(grade_matrix)

grade_matrix.to_csv("C:/users/desktop", encoding="utf-8", index=False)  # csv파일로 저장
grade_matrix.to_excel("C:/users/desktop", index=False)                  # 엑셀파일로 저장
3
opener = openpyxl.load_workbook("C:/users/desktop/grade_matrix.xlsx")    # 엑셀파일 열기
sheet = opener['Sheet1']

contents = {}                                            
for i in range(2, sheet.max_row +1) :              # 엑셀파일의 내용을 담을 배열
	grade = sheet.cell(row=i, columm =1).value
    name = sheet.cell(row=i, columm=2).value
    contents[grade] = name
print(contents)


# 간단하게 엑셀파일 읽기
data = pd.read_excel("C:/users/desktop/grade_matrix.xls", sheet_name = "Sheet1", encoding = "utf-8")

# 간단하게 csv 파일 읽기
file = open("C:/users/desktop/grade_matrix.csv", encoding="utf-8")
file_csv = csv.reader(file)
fir i in file_csv:
	print(i)