请编写一个Python函数，返回给定列表的中位数，如果列表长度为奇数，则返回中间元素，如果列表长度为偶数，则返回中间两个元素的平均值。
def find_median(lst): 
    # TODO: Find the median of lst 
    return median
print(find_median([1, 2, 3, 4, 5]))
print(find_median([1, 2, 3, 4, 5, 6]))
```python
def find_median(lst):
    sorted_lst = sorted(lst)
    length = len(sorted_lst)
    if length % 2 == 0:
        # for even list return mean of middle two elements
        return (sorted_lst[(length // 2) - 1] + sorted_lst[length // 2]) / 2
    else:
        # for odd list return middle element
        return sorted_lst[length // 2]

print(find_median([1, 2, 3, 4, 5])) # Output: 3
print(find_median([1, 2, 3, 4, 5, 6])) # Output: 3.5
```