Working with NumPy in ArcGIS

Numerical Python (NumPy) is a fundamental package for scientific computing in Python, including support for a powerful N-dimensional array object. NumPy provides an avenue to perform complex mathematical operations and has been part of the ArcGIS software installation since 9.2. For more information, see the NumPy websiteExternal link.

A Python NumPy array is designed to deal with large arrays. There are many existing Python functions that have been created to process NumPy arrays, the most noted being contained in the SciPy scientific computing package for Python.

Working with tables and feature data

Table and feature classes can be converted to and from NumPy arrays using functions in the data access (arcpy.da) module.

To convert NumPy arrays to tables and feature classes, the arrays must be structured arrays. Structured arrays include fields (or structs) that are used to map the data to field in ArcGIS table and feature classes. For more information on structured arrays, see Structured arraysExternal link.

Create a structured NumPy array.

import numpy

array = numpy.array([(471316.383, 5000448.782), (470402.493, 5000049.216)],
                    numpy.dtype([('X', '>f8'),('Y', '>f8')]))

Once created, a structured NumPy array can be converted to a feature class or table.

Convert a NumPy array to a geodatabase feature class.

import arcpy
import numpy

outFC = "C:/data/texas.gdb/fd/pointlocations"

# Create a numpy array with an id field, and a field with a tuple 
#  of x,y coordinates
array = numpy.array([(1, (471316.3835861763, 5000448.782036674)),
                     (2, (470402.49348005146, 5000049.216449278))],
                    numpy.dtype([('idfield', numpy.int32),('XY', '<f8', 2)]))

# Define a spatial reference for the output feature class
SR = arcpy.Describe("C:/data/texas.gdb/fd").spatialReference

# Export the numpy array to a feature class using the XY field to
#  represent the output point feature
arcpy.da.NumPyArrayToFeatureClass(array, outFC, ['XY'], SR)
arcpy.da functions for working with tables and feature data

Functions

Explanation

ExtendTable

Join the contents of a NumPy structured array to a table based on a common attribute field.

FeatureClassToNumPyArray

Convert a feature class to NumPy structured array.

NumPyArrayToFeatureClass

Convert a NumPy structured array to a feature class.

NumPyArrayToTable

Convert a NumPy structured array to a table.

TableToNumPyArray

Convert a table to NumPy structured array.

Integer fields in NumPy arrays don't support nulls. If data converted using FeatureClassToNumPyArray or TableToNumPyArray contains nulls, the rows containing the nulls should either be skipped entirely or masked with a substitute value.

Skip all records that include a null.

array = da.FeatureClassToNumPyArray(fc, fields, skip_nulls=True)

Mask Nones in integer fields with different values using a dictionary.

fields = ['field1', 'field2']
arcpy.da.FeatureClassToNumPyArray(fc, fields, null_value=-9999)

Type conversions

The dtypes of the created array are determined from the field type of the input table or feature class.

Field type

NumPy dtype

Single

numpy.float32

Double

numpy.float64

SmallInteger

numpy.int32

Integer

numpy.int32

OID

numpy.int32

GUID

<U64

String

<u1, <u10, and so on

NoteNote:

String fields converted to an array will have the same width. For instance, a string field with a width of 20 will have a dtype of <u20.

Other field types not listed above, including date, raster, and BLOB fields are not supported. Geometry fields are also not supported, but multiple geometry properties can be added to the array using the special tokens listed below.

Token

Description

SHAPE@XY

The feature's centroid x,y coordinates

SHAPE@TRUECENTROID

The feature's true centroid x,y coordinates

SHAPE@X

The feature's x-coordinate

SHAPE@Y

The feature's y-coordinate

SHAPE@Z

The feature's z-coordinate

SHAPE@M

The feature's m-value

SHAPE@AREA

The feature's area

SHAPE@LENGTH

The feature's length

Memory considerations

NumPy preallocates memory when creating arrays. An array that requires more memory than is available will fail with a MemoryError exception.

Tips to avoid MemoryError exceptions:

  • Delete array objects after use; deleting the array will release the memory.
  • Use only those fields you need, especially text fields; a text field converted to an array will consume 4 bytes for every character of width. For instance, a string field with a width of 100 will consume 400 bytes of memory for each value in the array.

NumPy dtype

Number of bytes per value

<U1

4

numpy.int32

4

numpy.float32

4

numpy.float64

8

NoteNote:

numpy.nbytes returns a dictionary of dtypes and number of bytes.

Working with rasters

Raster can be converted to and from NumPy arrays using the ArcPy functions RasterToNumPyArray and NumPyArrayToRaster. You may want to convert an ArcGIS raster to a NumPy array to:

  1. Implement one of the many existing Python functions that can be applied to a NumPy array (for example, run filters on the data, perform multidimensional analysis, or utilize optimization routines).
  2. Develop a custom function by accessing the individual cells within the NumPy array (for example, to implement neighborhood notation, change individual cell values, or run accumulative operators on an entire raster).

arcpy functions for working with rasters

Functions

Explanation

RasterToNumPyArray

Convert a raster to a NumPy array.

NumPyArrayToRaster

Convert a NumPy array to a raster.

A raster is converted to a NumPy array to calculate the percentage of the cell value in the entire raster row. A new raster is then created.

import arcpy
import numpy

my_array = arcpy.RasterToNumPyArray('C:/data/inRaster')
my_array_sum = my_array.sum(1)
my_array_sum.shape = (my_array.shape[0], 1)
my_array_perc = (my_array * 1.0) / my_array_sum
new_raster = arcpy.NumPyArrayToRaster(my_array_perc)
new_raster.save("C:/output/fgdb.gdb/PercentRaster")
4/12/2013