python numpy按列排序

python numpy按列排序

import random

import numpy as np

if __name__ == '__main__':

    data = np.array([[5, 2, 3], [1, 0, 6], [7, 0, 5]])

    # 使用 argsort 函数对第二列的值进行排序，并获取排序后的索引
    sorted_indices = np.argsort(data[:, 1])

    # 使用排序后的索引来排序整个数组
    sorted_data = data[sorted_indices]

    print(sorted_data)