.flatten()

Anonymous contributor's avatar
Anonymous contributor
Anonymous contributor's avatar
Anonymous contributor
Published Jul 8, 2024
Contribute to Docs

The .flatten() function in NumPy is a powerful tool for reshaping multi-dimensional arrays into one-dimensional arrays. This function simplifies complex arrays and makes them easier to work with.

Syntax

ndarray.flatten(order='C')
  • ndarray: Refers to the multi-dimensional array that is to be flattened.
  • order: The order in which the array is to be flattened. Here are the values that it can have:
    • C (Default): To flatten in row-major (C-style) order.
    • F: To flatten in column-major (Fortran-style) order.
    • A: To flatten in column-major order if the array is Fortran-contiguous in memory, otherwise in row-major order.
    • K: To flatten the array in the order the elements occur in memory.

Example

The following example demonstrates the usage of the .flatten() function:

import numpy as np
# Create a 2D array
arr = np.array([[1, 2, 3], [4, 5, 6]])
# Flatten the array
flattened_arr = arr.flatten()
# Print the flattened array
print(flattened_arr)

The above code generates the following output:

[1 2 3 4 5 6]

In this example, a 2D array arr is first created. Then, the .flatten() function is used to flatten the array into a one-dimensional array flattened_arr.

Codebyte Example

The following example shows the use of the .flatten() function:

Code
Output
Loading...

All contributors

Looking to contribute?

Learn Python:NumPy on Codecademy