Python 기초 2

✍🏻 배운점

 

파이썬으로 간단한 예제들을 통해 조건문을 공부할 수 있었다. 

또한, 파이썬을 사용하여 파일을 읽고 쓰며 폴더 작업에 대해서도 공부하여 데이터가 많은 작업에서도 쉽게 작업할 수 있었다.

실제 데이터들을 통해 앞서 배운 것들을 적용해보고 더 알아갈 수 있었다.

(특히, 맥 환경에서 주의할 점도 파악할 수 있어 다음에는 같은 실수를 하지 않을 수 있을 것 같다 :D)

 

 

📁 목차

1. 조건문 연습 1 : 사용자 입력에 따른 다른 결과 출력
2. 조건문 연습 2 : 윤년 확인
3. list comprehension (리스트 축약형)
4. multiLine
5. 노래 가사에서 가장 많이 등장한 단어 조회하기
6. 현재 작업 경로 확인하기
7. 절대 경로와 상대 경로
8. 자주 사용하는 폴더 작업 명령어
9. 파일 쓰기 및 읽기
10. 파일 옮기기
11. 성별에 따른 파일을 다른 폴더로 구분 후 성별에 따른 평균값 구하기
12. 스타벅스 매장 사진을 다른 폴더에 저장하고 해당 이미지 명을 매장명으로 저장하기

 

 

---

 

 

 

들어가며

 

지난 파이썬 기초 1을 리뷰한 후, 직접 예제 코드를 구현한 후 정답과 비교해보았다.

부족한 부분에 집중하고 신경써야 할 부분들에 대해 짚고 넘어가고자 한다.

 

 

 

 

 

조건문 연습 1 : 사용자 입력에 따른 다른 결과 출력

 

# 조건문 연습
# distance 라는 변수에는 상수 값이 들어갈 예정입니다.
# 이 distance 가 3km 이하로 작다면 "걸어 가세요" 라고 말하고
# 3km 보다 멀고 10 km 보다 작다면 "버스 타세요" 라고 말해야 합니다
# 만약 10km 이상이라면 "택시 타세요"라고 말해야 합니다
# distance 와 상관없이 "수고하셨습니다" 를 출력합니다

distance = input("input distance?")

distance = int(distance) # 문자열로 입력받기 때문에 int로 변환을 해주어야 함

if distance <= 3:
    print("걸어 가세요")
    
elif distance < 10:
    print("버스 타세요")
    
else:
    print("택시 타세요")

print("수고하셨습니다")

 

input 메서드를 통해 사용자에게 입력을 받는다.

이 때, input은 string 형이기 때문에 distance 의 값을 int 형과 비교 하기 위해서 int 로 형 변환을 해주어야 한다.

 

 

 

 

조건문 연습 2 : 윤년 확인

 

# 윤년 확인

for year in range(1900, 2101):
    if year % 4 == 0:
        if year % 100 == 0:
            if year % 400 == 0:
                print(f"{year} : 윤년")
            else:
                print(f"{year} : 평년")
        else:
            print(f"{year} : 윤년")

 

depth 가 3으로, 클린 코드 측면에서 비효율적인 나의 코드이다.

 

for year in range(1900, 2101):
	if year % 4 == 0 and year % 100 != 0 or year % 400 == 0:
    	print(f"{year} : 윤년")

 

이렇게 한 줄로 깔끔하게 코딩하는 법을 익혀야 한다.

 

 

 

list comprehension (리스트 축약형)

 

total = [x  for x in range(1, 5)]

# 15 배수 구하기 예시
total2 = [x  for x in range(1, 100) if x % 3 == 0 and x % 5 == 0]

 

복잡한 코드보다 이렇게 간단하게 한 줄로도 작성할 수 있다.

 

 

 

multiLine

 

multiline = """dddd
ddddd"""

 

multiline은 """ 으로 이루어지며 개행 문자인 '\n' 도 저장이 된다.

'

 

 

노래 가사에서 가장 많이 등장한 단어 조회하기 

 

ditto = '''Woo woo woo woo ooh
Woo woo woo woo
Stay in the middle
Like you a little
Don’t want no riddle
말해줘 say it back
Oh say it ditto
아침은 너무 멀어
So say it ditto
훌쩍 커버렸어
함께한 기억처럼
널 보는 내 마음은
어느새 여름 지나 가을
기다렸지 all this time
Do you want somebody
Like I want somebody
날 보고 웃었지만
Do you think about me now yeah
All the time yeah
All the time
I got no time to lose
내 길었던 하루
난 보고 싶어
Ra-ta-ta-ta 울린 심장 (Ra-ta-ta-ta)
I got nothing to lose
널 좋아한다고
wooah wooah wooah
Ra-ta-ta-ta 울린 심장 (Ra-ta-ta-ta)
But I don't want to
Stay in the middle
Like you a little
Don’t want no riddle
말해줘 say it back
Oh say it ditto
아침은 너무 멀어
So say it ditto
I don't want to
Walk in this 미로
다 아는 건 아니어도
바라던 대로
말해줘 Say it back
Oh say it ditto
I want you so, want you
So say it ditto
Not just anybody
너를 상상했지
항상 닿아있던
처음 느낌 그대로 난
기다렸지 all this time
I got nothing to lose
널 좋아한다고
wooah wooah wooah
Ra-ta-ta-ta 울린 심장 (Ra-ta-ta-ta)
But I don't want to
Stay in the middle
Like you a little
Don’t want no riddle
말해줘 say it back
Oh say it ditto
아침은 너무 멀어
So say it ditto
I don't want to
Walk in this 미로
다 아는 건 아니어도
바라던 대로
말해줘 Say it back
Oh say it ditto
I want you so, want you
So say it ditto
Woo woo woo woo ooh
Woo woo woo woo'''

 

ditto 라는 노래의 가사이다.

 

ditto = ditto.lower()

 

ditto 의 가사에는 대소문자가 섞여 있기 때문에 lower 메서드를 통해 모두 소문자로 바꾸어 준다.

 

dict_test = {}
temp = ditto.split();

for word in temp:
    
    if word in dict_test:
        dict_test[word] += 1
    else:
        dict_test[word] = 1

 

 

그 전 글에서 공부했던 대로, 빈 dictionary 를 만들고 split 메서드를 사용해 단어를 구분한다.

그 후, 이미 dict_test 안에 word 가 있다면 개수를 + 1, 없다면 1로 만들어준다.

 

 

def tmp(x):
	return x[1]

sorted(dict_test.items(), key = tmp, reverse = True) # key 부분을 기준으로 정렬해준다.

 

key 를 기준으로 내림차순 정렬하면, 가사에 가장 많이 등장한 순으로 각 단어의  등장 횟수를 알 수 있다.

 

 

 

현재 작업 경로 확인하기 

 

import os
os.getcwd() # 내가 지금 작업하는 경로

 

내가 지금 작업하는 현재 경로가 어디인지 알 수 있다.

 

 

 

 

 

 

 

절대 경로와 상대 경로

 

 

절대 경로란, 전체 경로를 의미하고, (예: c :\\workspace\\test)
상대 경로란, 현재 위치 '기준'으로의 경로를 의미한다. (예: .\test)

 

 

 

 

 

 

 

자주 사용하는 폴더 작업 명령어

 

os.mkdir("") # make directory
os.chdir("") # 다른 directory 로 change
os.listdir("") # 해당 directory 안에 있는 파일의 list

 

위 명령어들은 알아두면 유용하다.

 

 

 

 파일 쓰기 및 읽기 

 

f = open("./pic.txt", "w", encoding='utf-8')
for roots, dirs, files in os.walk("c:/Windows/"):
    for file in files:
        f.write(f"{roots}/{file}\n")
f.close()

 

파일을 쓰고("w") 

 

f = open('./이름_생년_성별_10000/', "r", encoding='utf-8')
f.close()

 

 

파일을 읽을("r") 수 있다.

 

 

 

 

 파일 옮기기

 

import shutil
shutil.move("./pic.txt", "./test/") # pic.txt 파일을 test 폴더로 옮기기

 

shutil 라이브러리를 사용하여 move 메서드를 통해 다른 곳에 있는 파일을 또 다른 폴더 안으로 위치를 바꿀 수 있다.

 

 

 

성별에 따른 파일을 다른 폴더로 구분 후 성별에 따른 평균값 구하기

 

 

### 문제 요구사항

# male 폴더 안에 1, 3 파일 옮기기
# female 폴더 안에 2, 4 파일 옮기기
# 파일을 읽어서 male 리스트, female 리스트에 각각 append 하기
# 평균 구하기

import shutil
import os

if os.path.isdir("./male") == False:
    os.mkdir("./male")
    
if os.path.isdir("./female") == False:
    os.mkdir("./female")


for file in os.listdir("./이름_생년_성별_10000/"):
    if file.split(".")[-1] == 'txt':
        if file.split("_")[2][0] == '1' or file.split("_")[2][0] == '3':
            shutil.move(f"./이름_생년_성별_10000/{file}", "./male/")
        else:
            shutil.move(f"./이름_생년_성별_10000/{file}", "./female/")
            

male = []

for file in os.listdir("./male/"):
    if file.split(".")[-1] == 'txt':
        f = open(f"./male/{file}", "r", encoding='utf-8')
        txt = f.read()
        txt = int(txt.replace(",", ""))

        male.append(txt)
        f.close()
        
        
female = []

for file in os.listdir("./female/"):
    if file.split(".")[-1] == 'txt':
        f = open(f"./female/{file}", "r", encoding='utf-8')
        txt = f.read()
        txt = int(txt.replace(",", ""))

        female.append(txt)
        f.close()


male_avg = sum(male) / len(male)
female_avg = sum(female) / len(female)


male_avg

female_avg

 

구현 자체는 어렵지 않은 문제였다.

 

그러나, 맥 환경은 새롭게 생성되는 파일들이 있기 때문에, 에러가 난다면 이를 조건문을 통해 걸러주는 로직을 항상 추가해야 한다고 한다.

(맥에서는 '.DS_Store' 라는 파일을 생성하는데, 전체 파일을 보다보면 이를 걸러주는 로직이 없어 에러가 발생하는 경우가 다분하니 주의해야 한다.)

 

조건문을 통해 파일 확장자명을 아래와 같이 조건에 꼭 포함시키도록 하자.

 

if file.split(".")[-1] == 'txt':

 

내가 살피고 싶은 파일 확장자명을 따로 지정해주는 조건문을 통해 원활한 진행이 가능하다.

 

 

 

스타벅스 매장 사진을 다른 폴더에 저장하고 해당 이미지 명을 매장명으로 저장하기

 

 

import requests
url = "https://www.starbucks.co.kr/store/getStore.do?r=GKYHQKU7D3"
payload = {"in_biz_cds" : "0",
"in_scodes" : "0",
"ins_lat" : "37.566535",
"ins_lng" : "126.9779692",
"search_text" : "",
"p_sido_cd" : "01",
"p_gugun_cd" : "",
"in_distance" : "0",
"in_biz_cd" : "",
"isError" : "true",
"searchType" : "C",
"set_date" : "",
"all_store" : "0",
"T03" : "0",
"T01" : "0",
"T27" : "0",
"T12" : "0",
"T09" : "0",
"T30" : "0",
"T05" : "0",
"T22" : "0",
"T21" : "0",
"T10" : "0",
"T36" : "0",
"T43" : "0",
"T48" : "0",
"P10" : "0",
"P50" : "0",
"P20" : "0",
"P60" : "0",
"P30" : "0",
"P70" : "0",
"P40" : "0",
"P80" : "0",
"whcroad_yn" : "0",
"P90" : "0",
"new_bool" : "0",
"iend" : "1000",
"rndCod" : "IZVHHSIFWC",}
r =  requests.post(url, data=payload)
star = r.json()['list']

 

이번에도 서울시 스타벅스 매장의 웹페이지에서 크롤링을 통해 진행한다.

 

 

# 스타벅스 매장 사진을 starbuck 폴더 생성하고
# 매장명.jpg 파일을 저장
from urllib import request
import shutil
import os

if os.path.isdir("./starbucks") == False:
    os.mkdir("./starbucks")

for x in star:
    if x['defaultimage'] != None:
        if x['defaultimage'].split(".")[-1] == 'jpg':
            request.urlretrieve("https://www.starbucks.co.kr" + x['defaultimage'], f"./starbucks/{x['s_name']}.jpg")
        elif x['defaultimage'].split(".")[-1] == 'png':
            request.urlretrieve("https://www.starbucks.co.kr" + x['defaultimage'], f"./starbucks/{x['s_name']}.png")
        elif x['defaultimage'].split(".")[-1] == 'JPG':
            request.urlretrieve("https://www.starbucks.co.kr" + x['defaultimage'], f"./starbucks/{x['s_name']}.JPG")

 

 

스타벅스 매장 사진을 그냥 다운로드 하게 될 경우,

 

지금 현재 작업하는 디렉토리에 저장되기 때문에 이미지를 저장할 새로운 폴더를 만들어 주었다.

 

그리고, 그냥 이미지를 다운 받으면 이미지의 이름이 숫자만으로 구성되어 어떤 매장인지 쉽게 이해할 수 없다.

 

따라서, 스타벅스 매장 이미지를 다운 받을 때, 매장명이 들어가도록 이미지 파일을 저장하도록 한다.