python/2차원 리스트

2차원 리스트[삭제]

june__Park 2021. 3. 1. 01:01
# 2차원배열[삭제]
# 이름과 번호를 입력받아 삭제하기

# 예)
# 삭제할 이름 입력 : 철수
# 삭제할 번호 입력 : 1
# 삭제 => ["철수", "김밥"]
#
# 삭제할 이름 입력 : 철수
# 삭제할 번호 입력 : 2
# 삭제 => ["철수", "바나나"]

jang = [
        ["철수", "소고기"],
        ["철수", "김밥"],
        ["영희", "김치"],
        ["철수", "바나나"],
        ["철수", "새우깡"],
        ["영희", "오징어"],
        ["영희", "맛밤"]
    ]

while True:
    print("[1] 삭제하기")
    print("[2] 전체보기")
    print("[3] 종    료")
    sel = int(input("입력: "))

    if sel == 1:
        name = input("삭제할 이름 입력: ")
        num = int(input("삭제할 번호 입력: "))
        i = 0
        cnt = -1
        del_idx = -1
        while i < len(jang):
            if jang[i][0] == name:
                cnt += 1
            if jang[i][0] == name and cnt == num:
                del_idx = i
                del(jang[del_idx])
                break
            i += 1
        if del_idx == -1:
            print("잘못된 입력입니다.")

    elif sel == 2:
        for i in range(len(jang)):
            print(jang[i])
    elif sel == 3:
        print("종료")
        break