Làm phẳng mảng numpy 2d thành mảng 1d trong Python

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

Với làm phẳng

Hàm flatten trong numpy là một cách trực tiếp để chuyển đổi mảng 2d sang mảng 1D.

Ví dụ

import numpy as np
array2D = np.array([[31, 12, 43], [21, 9, 16], [0, 9, 0]])
# printing initial arrays
print("Given array:\n",array2D)
# Using flatten()
res = array2D.flatten()
# Result
print("Flattened array:\n ", res)

Đầu ra

Chạy đoạn mã trên cho chúng ta 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 chức năng khác được gọi là ravel sẽ thực hiện một công việc tương tự là 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]])
# printing initial arrays
print("Given array:\n",array2D)
# Using ravel
res = array2D.ravel()
# Result
print("Flattened array:\n ", res)

Đầu ra

Chạy đoạn mã trên cho chúng ta 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]