How to use numpy.ndarray.flatten() in Python

How to use numpy.ndarray.flatten() in Python

In Python, numpy.ndarray.flatten() is a method that is used to flatten a multi-dimensional NumPy array into a one-dimensional array.

This method is often used to prepare data for machine learning algorithms or to make it easier to manipulate the data in a multi-dimensional array.

Syntax

Here is the syntax for numpy.ndarray.flatten():

numpy.ndarray.flatten(order='C')

This method takes the following argument:

• order (optional): The order in which the elements of the array are flattened. The default is 'C', which means that the elements will be flattened in row-major order (i.e., the last index will vary the fastest). You can also specify 'F' to flatten the array in column-major order (i.e., the first index will vary the fastest).

Examples

Here is an example of how you might use numpy.ndarray.flatten() to flatten a multi-dimensional array. Suppose you have the following array:

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6]])

You can use numpy.ndarray.flatten() to flatten this array into a one-dimensional array, like this:

a_flat = a.flatten()

This will result in the following array:

array([1, 2, 3, 4, 5, 6])

You can also specify the order argument to numpy.ndarray.flatten() to control the order in which the elements are flattened.

For example, you can flatten the array in column-major order like this:

a_flat = a.flatten(order='F')

This will result in the following array:

array([1, 4, 2, 5, 3, 6])