Numpy 常用操作

1. 排序

# 升序：
sort_order = np_data[:,col_index].argsort()
sort_data = np_data[sort_order]

# 降序：
sort_order = np_data[:,col_index].argsort()
sort_order = sort_order[::-1]
sort_data = np_data[sort_order]

# 举例：
np_data = np.array([[1.0, 8.1], [3.0, 7.3], [2.0, 5.2], [4.0, 1.4]])
sort_order = np_data[:,0].argsort()
sort_data = np_data[sort_order]
print(sort_data)

2. 去重

# np.unique 无论输入几维，总是返回一维升序numpy数组
# 对一列数据去重
np_unique = np.unique(np_data[:,col_index])

# 对所有数据去重
np_unique = np.unique(np_data)

3. 转换为 list

temp_list = np_data.tolist()

4. 数组合并

# 合并行（行数增加，要求a和b列数相同，合并后c的行数为a的行数+b的行数）
c = np.append(a, b, axis=0)

# 合并列（列数增加，要求a和b行数相同，合并后c的列数为a的列数+b的列数）
c = np.append(a, b, axis=1)