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

Cách tìm giao (intersection) giữa hai mảng NumPy


Trong bài này, tìm giao của hai mảng numpy. Giao là mảng chứa phần tử chung của cả hai mảng gốc.

Thuật toán

Bước 1: Import numpy.
Bước 2: Định nghĩa hai mảng numpy.
Bước 3: Tìm giao bằng numpy.intersect1d().
Bước 4: In mảng giao.

Mã ví dụ

import numpy as np
array_1 = np.array([1, 2, 3, 4, 5])
print("Array 1:\n", array_1)
array_2 = np.array([2, 4, 6, 8, 10])
print("\nArray 2:\n", array_2)
intersection = np.intersect1d(array_1, array_2)
print("\nGiao của hai mảng là:\n", intersection)

Kết quả

Array 1:  [1 2 3 4 5]
Array 2:  [2 4 6 8 10]
Giao của hai mảng là:  [2 4]