<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def vector_element(r, i, j):
 r"""Extract an matrix element of a vector operator. Matrix([ [x_{21}], [y_{21}], [z_{21}]]) """

    return Matrix([r[p][i, j] for p in range(3)])