- 5
- 이니스프리
- 조회 수 1195
SIR 디스달나노 님의 질문에 대한 파이썬 스크립트에요 ^-^
질문 내용 펌:
문제1)
$arr = ['A', 'A', 'B', 'B', 'A', 'B', 'A', 'C', 'C', 'C'];
무작위로 배열에 알파벳이 저장되어 있다.
다음과 같은 알파벳의 위치를 배열에 저장하시오.
단! 함수를 사용하여 배열의 리턴값을 받으시오.
$result = 함수($arr)
결과값 :
$result[0] = "A, 0, 1, 4, 6";
$result[1] = "B, 2, 3, 5";
$result[2] = "C, 7, 8, 9";
위 질문에 대해 파이썬으로 A~Z까지 일반적으로 작동하는 함수를 작성해봤네요~!
너무 오래 파이썬을 쉬면 감이 떨어져서 말이죠 ㅠㅠ
문제에서 아스키 코드를 이용하지 않는 풀이를 요구한다면 한참 길어질 것 같네요 ㄷㄷ
그리고 이런 유형의 문제를 풀 때에는 적절한 초기값의 리스트를 생성하는게 편하더군요 :)
def func(list): res = [] # A~Z까지 리스트 안의 리스트를 생성합니다. for i in range(26): res.append([chr(65 + i)]) # 각 문자의 위치를 리스트에 넣습니다. for idx1, l in enumerate(list): res[ord(l) - 65].append(idx1) # 리스트 안의 리스트의 길이가 1인 경우를 제외하여 새 리스트를 생성합니다. final = [x for x in res if len(x) != 1] return final data = ['A', 'A', 'B', 'B', 'A', 'B', 'A', 'C', 'C', 'C'] result = func(data) print(result)
결과 : [['A', 0, 1, 4, 6], ['B', 2, 3, 5], ['C', 7, 8, 9]]
마지막에 새 리스트를 생성하는 방식이 아니라 del이나 remove를 for문에서 돌리는 방식을 택하면
del이나 remove가 작동할 때마다 리스트의 길이가 짧아지기 때문에 제대로 된 결과를 얻을 수 없어요 ㅠㅠ
작성자
댓글 5
오오~ 빈 리스트에서 append를 할 수 없는 문제를 이렇게 if문으로 처리하면 되는군요 ㅎㄷㄷ
별표를 이용한 가변인자는 전혀 생각도 못했네요~!
역시 이런게 파이써닉한 코딩이군요!! ㄷㄷ
humit 님 덕분에 많이 배우고 가네요 ^-^
번번이 정말 감사드립니다~!
코로나 때문에 학사일정이 꼬이셨겠지만 이번 학기도 홧팅하시고 항상 건강하세요 :)
해당 자료형을 쓰면 따로 if 문으로 분기를 해주지 않아도 됩니다 :)
from collections import defaultdict
arr = ['A', 'A', 'B', 'B', 'A', 'B', 'A', 'C', 'C', 'C']
m = defaultdict(list)
for idx, item in enumerate(arr):
m[item].append(idx)
result = [[k, *v] for k,v in sorted(m.items())]
collections는 무궁무진하네요~!
즐거운 주말 되세요~ 감사합니다 :)
dictionary 자료형을 사용하는 방법도 있습니다 :)
arr = ['A', 'A', 'B', 'B', 'A', 'B', 'A', 'C', 'C', 'C']
m = {}
for idx, item in enumerate(arr):
if item in m:
m[item].append(idx)
else:
m[item] = [idx]
result = [[k, *v] for k,v in sorted(m.items())]