โปรแกรม Python หา Intersection ของสองลิสต์

หาส่วนร่วม (intersection) ของสองลิสต์ คืออิลิเมนต์ที่ปรากฏในทั้งสองลิสต์

ตัวอย่าง

List1: [1,2,3]
List2: [2,3,6]
Intersection: [2,3]

อัลกอริทึม

1. รับสองลิสต์
2. List comprehension: [x for x in A if x in B]

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

def intertwolist(A, B):
    return [i for i in A if i in B]

A = []; B = []
n = int(input("Enter the size of the List: "))
print("Enter first list:")
for _ in range(n): A.append(int(input()))
print("Enter second list:")
for _ in range(n): B.append(int(input()))
print("THE FINAL LIST IS:", intertwolist(A, B))

ผลลัพธ์

Enter the size of the List: 5
Enter first list: 12 23 45 67 11
Enter second list: 23 45 88 11 22
THE FINAL LIST IS: [23, 45, 11]