Skip to main content
Documents
basicsArray CreationArray OperationsArray Computation & AnalysisLinear AlgebraRandom ProbabilityData Input/Output & Conversion

NumPy arange()

The `numpy.arange()` function creates an array of equally spaced values within a defined interval. It is commonly used for generating sequences of numbers for array manipulation and analysis in scientific computing.

Usage

`numpy.arange()` is used to create a sequence of numbers in a specified range. It can define the start, stop (non-inclusive), and step size for the array values.

numpy.arange([start, ]stop, [step, ]dtype=None)
  • start (optional) defines the beginning of the interval. Defaults to 0 if not provided.
  • stop is the end of the interval but is exclusive.
  • step (optional) specifies the spacing between values. Can be negative for descending sequences.
  • dtype (optional) determines the data type of the output array.

Examples

1. Basic Usage

import numpy as np
array = np.arange(5)
print(array)

This creates an array with values from 0 to 4.

2. Specifying Start, Stop, and Step

import numpy as np
array = np.arange(2, 10, 2)
print(array)

An array is created with values starting at 2 up to 10 (exclusive) with a step of 2, resulting in [2, 4, 6, 8].

3. Using a Floating-Point Step

import numpy as np
array = np.arange(1.0, 5.0, 0.5)
print(array)

This example generates an array with floating-point values starting at 1.0 up to 5.0 with a step of 0.5, producing [1.0, 1.5, 2.0, 2.5, 3.0, 3.5, 4.0, 4.5].

4. Descending Sequence with Negative Step

import numpy as np
array = np.arange(10, 2, -2)
print(array)

This creates an array [10, 8, 6, 4] by specifying a negative step.

5. Specifying Data Type

import numpy as np
array = np.arange(5, dtype=np.float32)
print(array)

Here, dtype is set to np.float32, affecting the precision and storage of the array elements.

Tips and Best Practices

  • Avoid round-off errors. Be cautious with floating-point step values, as they can lead to precision issues.
  • Use linspace for a fixed number of elements. If you need a specific number of elements instead of a specific step size, consider using numpy.linspace().
  • Specify dtype for clarity. When necessary, explicitly define the dtype to ensure the array has the desired data type.
  • Check array length. Verify the length if using non-integer steps, as the result may not include the endpoint.
  • Memory considerations. Be mindful of the potential for creating arrays with a very large number of elements, which can have significant memory implications. Note that platform-dependent behavior may affect the maximum size of an array.