Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

โปรแกรม Python จัดการลิสต์ด้วยคลาส (เพิ่ม ลบ แสดง)

บทนำ

ตัวอย่างการสร้างคลาสสำหรับจัดการลิสต์: เพิ่ม element, ลบ element, และแสดงลิสต์ พร้อมเมนูโต้ตอบบนคอนโซล

ตัวอย่างโค้ด

class ListManager:
    def __init__(self):
        self.items = []

    def add(self, value):
        self.items.append(value)
        print(f"เพิ่ม {value} แล้ว")

    def remove(self, value):
        try:
            self.items.remove(value)
            print(f"ลบ {value} แล้ว")
        except ValueError:
            print(f"ไม่พบ {value} ในลิสต์")

    def display(self):
        print("ลิสต์ปัจจุบัน:", self.items)
        return self.items

# โปรแกรมหลัก
manager = ListManager()

while True:
    print("\n0. ออก")
    print("1. เพิ่มข้อมูล")
    print("2. ลบข้อมูล")
    print("3. แสดงลิสต์")
    choice = input("เลือก: ")

    if choice == '1':
        val = int(input("ป้อนตัวเลข: "))
        manager.add(val)
    elif choice == '2':
        val = int(input("ป้อนตัวเลขที่จะลบ: "))
        manager.remove(val)
    elif choice == '3':
        manager.display()
    elif choice == '0':
        print("ออกจากโปรแกรม")
        break
    else:
        print("เมนูไม่ถูกต้อง")

ผลลัพธ์ตัวอย่าง

0. ออก
1. เพิ่มข้อมูล
2. ลบข้อมูล
3. แสดงลิสต์
เลือก: 1
ป้อนตัวเลข: 34
เพิ่ม 34 แล้ว
ลิสต์ปัจจุบัน: [34]
...
เลือก: 0
ออกจากโปรแกรม

คำอธิบาย

  • คลาส ListManager ห่อหุ้มลิสต์และพฤติกรรมต่างๆ (Encapsulation)
  • add() ใช้ append(), remove() ใช้ remove() (ลบตัวแรกที่เจอ)
  • try-except จัดการกรณีลบค่าที่ไม่มีในลิสต์
  • ลูป while True พร้อม break สำหรับเมนู

ปรับปรุง: รับข้อมูลหลายชนิด

def add(self, value):
    self.items.append(value)
    # ไม่ต้องแปลง int รับได้ทุกชนิด