天天看点

NumPy二维数组-按某列排序

1 a[a[:,0].argsort()]

取某列排序之后的索引,然后在原数组中按索引取出数据

NumPy二维数组-按某列排序
NumPy二维数组-按某列排序

2 转为pandas DataFrame

最好的方式是转为 DataFrame 之后再排序

# 按第二列排序,随后是第五列
pd.DataFrame(a).sort_values(by=[2,5]).to_numpy()

# 按列从左到右
pd.DataFrame(a).sort_values().to_numpy()