如何绘制分组数据?

2024-05-13 15:16:29 发布

您现在位置:Python中文网/ 问答频道 /正文

我有下面的数据框,它是一个200行的子集,有18种类型的汽车

import pandas as pd

cars = {'Vnum': [1497, 1923, 1002, 1229, 1168, 1644, 2002, 1879, 1265, 1176, 1305, 1080],
       'name': ['Honda Civic','Toyota Corolla','Honda Civic', 'Toyota Corolla','Ford Focus','Audi A4','Honda Civic','Honda Civic','Toyota Corolla','Toyota Corolla','Toyota Corolla','Honda Civic'],
       'Enum': [23, 9, 2, 45, 13, 4, 25, 11, 6, 14, 27, 8],
       'Syear': [2019, 2000, 2003, 2000, 2000, 2019, 1977, 2000, 2003, 2003, 2000, 2000],
      
       }

df = pd.DataFrame(cars, columns = ['Vnum', 'name','Vnum','Syear'])

print (df)

我需要你的帮助来可视化数据

我想要一个横轴上有汽车名字的条形图,还有一个显示每年汽车数量的条形图。(我添加了一个图像) 任何其他情节将工作,我只需要看看汽车和年数的关系

我不能离开这个密码

df_gg=df.groupby(['name','Syear']); df_gg.groups

enter image description here


Tags: 数据namedfcars汽车子集pd条形图
3条回答

您可以将分组的数据帧转换为数据透视表并进行打印

df = df.groupby(['name', 'Syear']).count().reset_index()

pd.pivot_table(df, index = 'name', columns = 'Syear', values = 'Vnum').plot(kind = 'bar', rot = 0)
plt.show()

enter image description here

尝试增加附加值

    import plotly.graph_objects as go
    
    df['name_year'] = df.name + ' ' + df.Syear.apply(str)
    
    traces = []
    for name_year in list(set(df.name_year)):
        trace = go.Bar(name=name_year, 
                x=df[df['name_year'] == name_year].name,
                y=df[df['name_year'] == name_year].Vnum,
                text=df[df['name_year'] == name_year].Syear,
                textposition='auto')
        traces.append(trace)
    
    fig = go.Figure(data=traces)
    fig.update_layout(barmode='group')
    fig.show()

enter image description here

更新: 固定代码

使用(有效替代方案)—

df.groupby(['name', 'Syear'])['Vnum'].count().unstack(level=-1).plot(kind = 'bar', rot = 0)

enter image description here

计时

@Mlang's溶液-

300 ms ± 59.2 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)

这个-

53.1 ms ± 4.65 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)

相关问题 更多 >