Obtain median values for arrays of different sizes

I understand that median is easy to get in an array of the same size.

For example, A = [1,2,3,4,5,6,7,8,9] and if you want to break 3 each to get the median.

median = np.median(A.reshape(3,3), axis=1)
print(median)
>> [2. 5. 8.]

You could do this.

But if you want to cut the number differently, for example, cut A into 2, 3, and 4.

median = func(A)
print(median)
>> [1.5, 4, 7.5]

can do this? I want to not spend any iteration.