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

Làm Phẳng Mảng 2D NumPy Thành 1D (Flatten a 2d Numpy Array into 1d Array) trong Python


Mảng 2D NumPy là mảng của các mảng. Trong bài này ta sẽ xem cách làm phẳng nó để có các phần tử dưới dạng mảng một chiều.

Với flatten

Hàm flatten trong NumPy là cách trực tiếp để chuyển mảng 2D thành mảng 1D.

Ví dụ

import numpy as np array2D = np.array([[31, 12, 43], [21, 9, 16], [0, 9, 0]]) # in mảng ban đầu print("Given array:\n",array2D) # Sử dụng flatten() res = array2D.flatten() # Kết quả print("Flattened array:\n ", res)

Đầu ra

Chạy mã trên cho kết quả sau:

Given array: [[31 12 43] [21 9 16] [ 0 9 0]] Flattened array: [31 12 43 21 9 16 0 9 0]

Với ravel

Có một hàm khác gọi là ravel cũng làm tương tự việc làm phẳng mảng 2D thành 1D.

Ví dụ

import numpy as np array2D = np.array([[31, 12, 43], [21, 9, 16], [0, 9, 0]]) # in mảng ban đầu print("Given array:\n",array2D) # Sử dụng ravel res = array2D.ravel() # Kết quả print("Flattened array:\n ", res)

Đầu ra

Chạy mã trên cho kết quả sau:

Given array: [[31 12 43] [21 9 16] [ 0 9 0]] Flattened array: [31 12 43 21 9 16 0 9 0]