You can try klongpy, a K-like array language implementation that runs atop numpy: https://pypi.org/project/klongpy/