Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách tìm hiệu tập hợp (set difference) giữa hai mảng NumPy


Trong chương trình này, tìm hiệu tập hợp của hai mảng numpy. Dùng setdiff1d() trong numpy. Hàm nhận array1, array2 và trả về giá trị duy nhất trong array1 không có trong array2.

Thuật toán

Bước 1: Import numpy.
Bước 2: Định nghĩa hai mảng numpy.
Bước 3: Tìm hiệu tập hợp bằng setdiff1d().
Bước 4: In kết quả.

Mã ví dụ

import numpy as np
array_1 = np.array([2, 4, 6, 8, 10, 12])
print("Array 1: \n", array_1)
array_2 = np.array([4, 8, 12])
print("\nArray 2: \n", array_2)
set_diff = np.setdiff1d(array_1, array_2)
print("\nHiệu tập hợp array_1 - array_2 là:\n", set_diff)

Kết quả

Array 1: [ 2  4  6  8 10 12]
Array 2: [ 4  8 12]
Hiệu tập hợp array_1 - array_2 là: [ 2  6 10]

Giải thích

Array 1 có phần tử 2, 6, 10 không có trong Array 2. Do đó [2 6 10] là hiệu tập hợp.