utils
utils¶
Functions that make some features in Matplotlib a bit more convenient.
- figure(figsize=1, *args, **kwargs)[source]¶
Matplotlib figure but a scalar figsize will multiply rcParams figsize.
Examples
The figure size of the two figures will be equivalent:
plt.rcParams['figure.figsize'] = [6.4, 4.8] fig1 = figure(2) fig2 = plt.figure(figsize=(12.8, 9.6))
- indexer(arr, index_name=None, axis=0)[source]¶
Utility function for when you want to index an array as part of an interactive function. For example:
iplt.plot(indexor(arr), idx = np.arange(5))
- Parameters
- arrarraylike
The array to be indexed
- index_namestr, default: None
The name of the variable the returned function should expect to receive. If
None
then the function will check foridx
,index
,ind
, andindx
.- axisint
which axis to index, default to first axis
- Returns
- ffunction
Function to be passed as an argument to an interactive plotting function
- nearest_idx(array, value, axis=None)[source]¶
Return the index of the array that is closest to value. Equivalent to
np.argmin(np.abs(array-value), axis=axis)
.- Parameters
- arrayarraylike
- valueScalar
- axisint, optional
From np.argmin: “By default, the index is into the flattened array, otherwise along the specified axis.”
- Returns
- idxIndexArray
index of array that most closely matches value. If axis=None this will be an integer