[프로그래머스 Python] 코딩 기초 트레이닝 Day 24 - 조건문, 반복문 활용, 이차원 리스트(배열)

---

1. 커피 심부름

팀의 막내인 철수는 아메리카노와 카페 라테만 판매하는 카페에서 팀원들의 커피를 사려고 합니다. 아메리카노와 카페 라테의 가격은 차가운 것과 뜨거운 것 상관없이 각각 4500, 5000원입니다. 각 팀원에게 마실 메뉴를 적어달라고 하였고, 그 중에서 메뉴만 적은 팀원의 것은 차가운 것으로 통일하고 “아무거나”를 적은 팀원의 것은 차가운 아메리카노로 통일하기로 하였습니다.
각 직원이 적은 메뉴가 문자열 배열 order로 주어질 때, 카페에서 결제하게 될 금액을 return 하는 solution 함수를 작성해주세요. order의 원소는 아래의 것들만 들어오고, 각각의 의미는 다음과 같습니다.

order의 원소	의미
“iceamericano”, “americanoice”	차가운 아메리카노
“hotamericano”, “americanohot”	따뜻한 아메리카노
“icecafelatte”, “cafelatteice”	차가운 카페 라테
“hotcafelatte”, “cafelattehot”	따뜻한 카페 라테
“americano”	아메리카노
“cafelatte”	카페 라테
“anything”	아무거나

제한사항

• 1 ≤ order의 길이 ≤ 1,000

코드 작성

def solution(order):
    menu = {'americano': 4500, 
            'cafelatte': 5000, 
            'anything': 4500}
    
    # hot, ice 제거하기(가격동일)
    remove_hot = [o.replace('hot', '')  for o in order]
    remove_sub = [o.replace('ice', '')  for o in remove_hot]
    return sum([menu[o] for o in remove_sub])

2. 그림 확대

직사각형 형태의 그림 파일이 있고, 이 그림 파일은 1 × 1 크기의 정사각형 크기의 픽셀로 이루어져 있습니다. 이 그림 파일을 나타낸 문자열 배열 picture과 정수 k가 매개변수로 주어질 때, 이 그림 파일을 가로 세로로 k배 늘린 그림 파일을 나타내도록 문자열 배열을 return 하는 solution 함수를 작성해 주세요.

제한사항

• 1 ≤ picture의 길이 ≤ 20
• 1 ≤ picture의 원소의 길이 ≤ 20
• 모든 picture의 원소의 길이는 같습니다.
• picture의 원소는 ‘.’과 ‘x’로 이루어져 있습니다.
• 1 ≤ k ≤ 10

코드 작성

def solution(picture, k):
    zoom_in = []
    for pixel in picture:
        new_pixel = ''
        for p in pixel:
            new_pixel += p*k
        zoom_in.append(new_pixel)
    return [s for s in zoom_in for _ in range(k)]

k배만큼 확대한 픽셀을 반환하기 위해 picture에 담긴 각 픽셀의 문자열을 k배만큼 new_pixel 변수에 넣어 zoom_in 이라는 리스트에 넣어주었다. 그리고 k배로 확대한 픽셀이 k개 존재해야하기 때문에 리스트컴프리헨션으로 각 픽셀을 k개씩 넣어주었다.

3. 조건에 맞게 수열 변환하기 3

정수 배열 arr와 자연수 k가 주어집니다.
만약 k가 홀수라면 arr의 모든 원소에 k를 곱하고, k가 짝수라면 arr의 모든 원소에 k를 더합니다.
이러한 변환을 마친 후의 arr를 return 하는 solution 함수를 완성해 주세요.

제한사항

• 1 ≤ arr의 길이 ≤ 1,000,000
  • 1 ≤ arr의 원소의 값 ≤ 100
• 1 ≤ k ≤ 100

코드 작성

def solution(arr, k):
    return [n+k for n in arr] if k % 2 == 0 else [m*k for m in arr]

4. l로 만들기

알파벳 소문자로 이루어진 문자열 myString이 주어집니다. 알파벳 순서에서 “l”보다 앞서는 모든 문자를 “l”로 바꾼 문자열을 return 하는 solution 함수를 완성해 주세요.

제한사항

• 1 ≤ myString ≤ 100,000
  • myString은 알파벳 소문자로 이루어진 문자열입니다.

코드 작성

def solution(myString):
    return ''.join(['l' if s < 'l' else s for s in myString])

5. 특별한 이차원 배열 1

정수 n이 매개변수로 주어질 때, 다음과 같은 n × n 크기의 이차원 배열 arr를 return 하는 solution 함수를 작성해 주세요.

• arr[i][j] (0 ≤ i, j < n)의 값은 i = j라면 1, 아니라면 0입니다.

제한사항

• 1 ≤ n ≤ 100

코드 작성

def solution(n):
    arr = [[0 for i in range(n)] for j in range(n)]
    for k in range(n):
        arr[k][k] = 1
    return arr

코드 작성(Numpy)

import numpy as np

def solution(n):
    return np.eye(n).tolist()

📍 문제 출처

• 커피 심부름
• 그림 확대
• 조건에 맞게 수열 변환하기 3
• l로 만들기
• 특별한 이차원 배열 1

👩🏻‍💻개인 공부 기록용 블로그입니다
오류나 틀린 부분이 있을 경우 댓글 혹은 메일로 따끔하게 지적해주시면 감사하겠습니다.