Update phonebook_simple.py

nicecoding1 · web-flow · commit f9c2d3b13a2c · 2022-01-09T12:37:59.000+09:00
diff --git a/phonebook_simple.py b/phonebook_simple.py
@@ -74,20 +74,18 @@
             print("입력값이 부족합니다. 다시 입력해주세요.")
             continue
 
-        tmp_list = []
+        del_cnt = 0
 
         for i in phonebook:
-            if not i[1] == a:
-                tmp_list.append([i[0], i[1]])
-
-        if len(tmp_list)>0:
-            phonebook.clear()
-            phonebook = tmp_list
+            if i[1] == a:
+                phonebook.remove([i[0], i[1]])
+                del_cnt += 1
 
+        if del_cnt > 0:
             print("{} 삭제 성공".format(a))
         else:
             print("{} 번호가 존재하지 않습니다.".format(a))
 
         
     elif m == "5":# 종료
-        break
+        break
