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. ArcGIS for Desktop applications include NumPy version 1.9.2. For more information, see the NumPy website.

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 arrays.

Create a structured NumPy array.

import numpy

arr = 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

out_fc = 'C:/data/texas.gdb/fd/pointlocations'

# Create a numpy array with an id field, and a field with a tuple 
#  of x,y coordinates
arr = 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
spatial_ref = 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(arr, out_fc, ['XY'], spatial_ref)

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.

arr = arcpy.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
Date	<M8[us]

Note:

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 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

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

Note:

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")