Numpy

IO

from io import StringIO

import numpy as np


c = StringIO("0 1\n2 3")

np.loadtxt(c)

Create array

np.arrange(1, 5, 1)

np.random.uniform(-1, 1, 100)

Create a new dimension from existing arrays

np.stack((arr1, arr2), axis=0)


u = np.array([[4.0, 2.0, 0.0, 0.0])

lat = np.random.randint(-90, high=90, size=10)



def ceil_to_0p25(a: float) -> float:

    return np.ceil(a / 0.25) * 0.25


def floor_to_0p25(a: float) -> float:

    return np.floor(a / 0.25) * 0.25

 Value counts

np.unique(arr, return_counts=True)

 expand array by 1

np.pad(arr, 1)

Testing / compare numpy arrays

np.array_equal(arr1, arr2)

numpy vectorize


np.array([360, 90, 180, 270])