[SPARK-53641][DOCS] Add PARTITION BY support in Arrow Python UDTF docs

### What changes were proposed in this pull request?

This PR adds PARTITION BY and ORDER BY semantics in Arrow Python UDTF docs. It also add more examples for Arrow Python UDTFs.

### Why are the changes needed?

To improve documentations

### Does this PR introduce _any_ user-facing change?

No

### How was this patch tested?

manual tests

### Was this patch authored or co-authored using generative AI tooling?

Yes

Closes #52392 from allisonwang-db/spark-53641-udtf-partition-by-docs.

Authored-by: Allison Wang <[email protected]>
Signed-off-by: Ruifeng Zheng <[email protected]>

Author: allisonwang-db