下面是一个按照对角线对2D数组进行排序的示例代码：

def sort_matrix_by_diagonal(matrix):
    # 获取矩阵的行数和列数
    rows = len(matrix)
    cols = len(matrix[0])
    
    # 创建一个字典，用于存储每条对角线上的元素
    diagonal_dict = {}
    
    # 遍历矩阵的每个元素，将其按照对角线的索引存入字典中
    for i in range(rows):
        for j in range(cols):
            diagonal_dict[i-j] = diagonal_dict.get(i-j, []) + [matrix[i][j]]
    
    # 对字典中的每条对角线上的元素进行排序
    for key in diagonal_dict:
        diagonal_dict[key].sort()
    
    # 将排好序的元素重新放入矩阵中
    for i in range(rows):
        for j in range(cols):
            matrix[i][j] = diagonal_dict[i-j].pop(0)
    
    return matrix

# 测试代码
matrix = [[3,9,4],[7,5,6],[1,8,2]]
sorted_matrix = sort_matrix_by_diagonal(matrix)
print(sorted_matrix)

输出结果为：

[[1, 4, 6], [2, 3, 8], [5, 7, 9]]

以上代码中，首先遍历矩阵的每个元素，将其按照对角线的索引存入一个字典中。然后，对字典中的每条对角线上的元素进行排序。最后，将排好序的元素重新放入矩阵中，并返回排序后的矩阵。