Сортировка списков списка

У меня есть список, содержащий числа. Я хочу отсортировать один из списков, а затем убедиться, что другие списки расположены в порядке отсортированного списка. Все списки имеют одинаковый размер.

import numpy as np

# Example list of lists with numbers
list_of_lists = [[5, 8, 9], [1, 4, 7], [6, 3, 2]]

# Convert lists to NumPy arrays
arrays = [np.array(lst) for lst in list_of_lists]

# Sort the first array
sorted_array = np.sort(arrays[0])

# Initialize a list to hold the reordered arrays
reordered_arrays = [sorted_array]  # Start with the sorted array

# Function to reorder an array based on the sorted positions of its elements
def reorder_array(arr, sorted_positions):
    reordered_arr = np.full_like(sorted_array, fill_value=np.nan)  # Initialize with NaNs
    for pos, val in enumerate(sorted_positions):
        if val in arr:
            idx = np.where(arr == val)[0][0]  # Find the index of the current value in the unsorted array
            reordered_arr[pos] = arr[idx]  # Place the value from the original array at the correct position
    return reordered_arr

# Reorder the other arrays
for arr in arrays[1:]:
    sorted_positions = np.argsort(arr)  # Find the sorted positions of elements in the current array
    reordered_arr = reorder_array(arr, sorted_positions)
    reordered_arrays.append(reordered_arr)


print(reordered_arrays)