Numpy Create Index Array Python Example