[파이썬] 리스트 관련 검색, 추가, 삭제 구현

1.데이터 검색 및 검색위치 반환

color = ["red", "orange", "yellow", "green", "blue", "indigo", "purple"]

# 데이터 검색 및 검색위치 반환
def is_on_list_color(a, b):
  return b in a
def index_of_list(a, b):
  return a.index(b)

x = input("검색할 색상을 입력하시오 : ")

if is_on_list_color(color, x) == True:
  print(f"{x} index number is ", index_of_list(color, x),)
else:
  print(f"sorry, {x} is not exist")

2.데이터 추가

def input_on_list(a, b):
  return a.append(b)
  
x = input("추가할 색상을 입력하시오 : ")
input_on_list(color, x)
print(color)

3. 데이터 삭제

def delete_on_list(a, b):
  return a.remove(b)  

x = input("삭제할 색상을 입력하시오 : ")
delete_on_list(color, x)
print(color)