Pandas实用技能,将列(column)排序的几种方法

  • Post author:
  • Post category:其他


来源:Python数据之道 (ID:PyDataLab)

作者:阳哥

大家好,我是阳哥。

Pandas 可以说是 在Python数据科学领域应用最为广泛的工具之一。

Pandas是一种高效的数据处理库,它以

dataframe



series

为基本数据类型,呈现出类似excel的二维数据。

在数据处理过程中,咱们经常需要将列按照一定的要求进行排序,以方便展示。

这里,阳哥来给大家分享下 在 Pandas 中将列排序的几种常用方法。

数据准备

文中主要使用了

pandas



akshare

,首先导入 Python 库,如下:

import pandas as pd
import akshare as ak

print(f'pandas version: {pd.__version__}')

本次使用的数据如下:

data = {
    'brand':['Python数据之道','价值前瞻','菜鸟数据之道','Python','Java'],
    'B':[4,6,8,12,10],
    'A':[10,2,5,20,16],
    'D':[6,18,14,6,12],
    'years':[4,1,1,30,30],
    'C':[8,12,18,8,2],
}

df = pd.DataFrame(data=data)
df

现将现有的

columns

输出,方便后面 copy 使用。

df.columns

# Index(['brand', 'B', 'A', 'D', 'years', 'C'], dtype='object')

Method 1

第一种方法,也是我自己常用的方法,就是自己将列的名称按需要进行手动排序,然后运行代码如下:

Method 2

第二种方法,是使用

.iloc

方法,通过列的位置来进行排序,如下:

Method 3

第三种方法,是使用

.loc

方法,通过列的名称来进行排序,如下:

这种方法跟第一种方法类似,个人觉得第一种方法更简洁些。

Method 4

第四种是 逆序 排序,算是排序中一种特定的排序方式。

# Method 4 ,逆序

cols = list(df.columns)

cols.reverse()

df[cols]

上述代码中,

cols.reverse()

是将列表(list)进行逆序排序。

此外,列表(list)的逆序排序,还可以用

cols[::-1]

来实现。因此,下面的方法也可以实现逆序排序。

# Method 4 ,逆序

cols = list(df.columns)

df[cols[::-1]]

关于 Python数据类型 list 的介绍,详细内容可以前往下面的链接:

实战案例:自由排序

有时候,当存在变量、列的数量较多,或者不同的dataframe中列的名称不完全一致等情况出现时,咱们不一定会通过列名称来实现排序。

这里分享一个实战案例,是关于制作基金的十大持仓数据表的,具体过程我就不在这里描述了,下面给出实现的函数,有兴趣的同学可以研究下。

自定义函数如下:

# 需要安装 akshare
# pip install akshare

years = ['2019','2020','2021']

def fund_stock_holding(years,code):
    data = pd.DataFrame()
    for yr in years:
        df_tmp = ak.fund_em_portfolio_hold(code=code,year=yr)
        data = data.append(df_tmp)

    data['季度']=data['季度'].apply(lambda x:x[:8])
    data['占净值比例'] = pd.to_numeric(data['占净值比例'])
    data = data.sort_values(['季度','持仓市值'],ascending=[True,False])
    df = data.set_index(['序号','季度']).stack().unstack([1,2]).head(10)
    df = df.loc[:,(slice(None), '股票名称')]
    df = df.droplevel(None,axis=1)
    df.columns.name=None
    df = df.reset_index()
#     df.index.name = None
    df['基金代码'] = code
    return df
    
df = fund_stock_holding(years,'005669')
df

得到的数据表格如下:

上面的表格中,我需要将

基金代码

这一列移动到

序号

这列的后面,由于

years = ['2019','2020','2021']

这是一个变量,当具体的值不同时,会导致列名称不一样,因此,在这种情况下我们不能直接使用列的具体名称,但咱们可以通过 列的位置组合来实现,列的调整具体如下:

    cols = df.columns.tolist()
    cols = cols[:1] + cols[-1:] + cols[1:-1]  # 将基金代码列名放前面
    df = df[cols]

将上面的调整过程整合到自定义函数中,完整的代码如下:

# 需要安装 akshare
# pip install akshare

years = ['2019','2020','2021']

def fund_stock_holding_update(years,code):
    data = pd.DataFrame()
    for yr in years:
        df_tmp = ak.fund_em_portfolio_hold(code=code,year=yr)
        data = data.append(df_tmp)

    data['季度']=data['季度'].apply(lambda x:x[:8])
    data['占净值比例'] = pd.to_numeric(data['占净值比例'])
    data = data.sort_values(['季度','持仓市值'],ascending=[True,False])
    df = data.set_index(['序号','季度']).stack().unstack([1,2]).head(10)
    df = df.loc[:,(slice(None), '股票名称')]
    df = df.droplevel(None,axis=1)
    df.columns.name=None
    df = df.reset_index()
#     df.index.name = None
    df['基金代码'] = code
    cols = df.columns.tolist()
    cols = cols[:1] + cols[-1:] + cols[1:-1]  # 将基金代码列名放前面
    df = df[cols]
    return df
    
df = fund_stock_holding_update(years,'005669')
df

效果如下:

当然,我最后实现的效果是将基金代码换成基金名称,这个可以想办法实现,效果如下:

小结

以上就是关于 Pandas 中 列名称排序的介绍,看似很简单的内容,在最后的实践中,也还是有些小技巧的。

欢迎大家来畅聊,Pandas 中有哪些实用的小技巧~~


大家读完顺手点下右下角的

“在看”

,就是最大的鼓励和支持了。



版权声明:本文为lemonbit原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。