Python Efficient Double Loop