correlation between geographical areas

I have an ongoing project with the following assignment:

Given data in csv files for ocean flows in two sets of vectors over a lapse of time.

Therefore I have, let’s say U vectors for latitudinal and V vectors for longitudinal areas over time.

I’m working in Python using Jupyter Notebook and numpy.

My U and V and subsequent speed (magnitude of vectors) have shapes 544 x 550 x 100.

100 is the time index, 550 is the x axis and 544 the y axis.

How do I find highest correlation between random areas of the map using only U?

I’ve tried using numpy.corrcoef, but the function takes only 2 dimensional array and not 3 dimensional.

Source: Python Questions