Fast small arrays in Python