Python NumPy repeat example