Array programming with NumPy