Tetris Python Code Without Loops In R