[파이썬] 입력된 리스트 요소값에 따른 MSG 출력 코드 단순화 정리

 

import random

a=random.randint(1,20)
b=random.randint(1,20)
c=random.randint(1,20)
lst=[a,b,c]
print(lst)

x=int(input("원하는 유전자 정보번호를 입력해주세요:"))

idx = lst.index(x)
print(lst[idx])
if lst[idx] == a:
    print("탈모에 관한 유전자입니다")
elif lst[idx] == b:
    print("백색증에 관한 유전자입니다")
elif lst[idx] == c:
    print("페닐케톤에 관한 유전자입니다")

if lst[idx]>=10:
    print("윗 질환 발병확률이 높습니다")
else: 
    print("윗 질환 발병확률이 현저히 낮습니다")

(방법)

import random

a=random.randint(1,20)
b=random.randint(1,20)
c=random.randint(1,20)
lst=[a,b,c]
# a, b, c에 맞는 출력 문자열을 리스트로 만듭니다.
msg = ['탈모에 관한 유전자입니다'
       ,'백색증에 관한 유전자입니다'
       ,'페닐케톤에 관한 유전자입니다']

print(lst)
x=int(input("원하는 유전자 정보번호를 입력해주세요:"))

idx = lst.index(x)     # 입력된 x로 lst에서 해당 요소의 인덱스를 찾습니다.
print(msg[idx])        # 해당 인덱스의 msg 문자열을 출력 합니다.

if lst[idx]>=10:
    print("윗 질환 발병확률이 높습니다")
else: 
    print("윗 질환 발병확률이 현저히 낮습니다")