Pandas groupby两列并创建一个计数总计的图

Question

我是Pandas的新手，我正在寻找一种方法来绘制按两列分组的数据。这是我的榜样：

首先，我按照“日期”（年份）和“主要类型”排序＆＃39;列。

Activity

现在，我可以得到一系列基本上我想要绘制的内容。

groups = df.groupby([df['Date'].map(lambda x: x.year), pri_type['Primary Type']])

但是当我绘制这个时，我得到一个非常混乱的情节，在x轴上有数千个标签。我想得到的是x轴上有日期的图和带有所有主要类型的ledgend。与此图类似的东西：

提前致谢！

Answer 1

你希望在x轴上显示什么，日期？如果是这样，您可以将日期设置为索引：     groups.set_index（＆＃39;日期＆＃39）

Answer 2

我想出的解决方案是将系列转换为数据框并使用unstack()方法。这是我做的：

# convert to a dataframe
df = groups.size().to_frame()

|       |               |  0
|------ | --------------|------
|Date   | Primary Type  |
|       | ARSON         | 1010
|       | ASSAULT       | 31384
| 2001  | BATTERY       | 93234
|       | BURGLARY      | 26031
|       | CRIM SEXUAL AS| 1723

# unstack() to pivot the data which puts it in the correct format for plot()
df.unstack(level=-1)

|            |0                    
|------------|-------|---------|-------...
|Primary Type|ARSON  |ASSAULT  |BATTERY...
|Date        |       |         |       ...
|2001        |1010.0 |31384.0  |93234.0...
|2002        |2938.0 |31993.0  |94235.0...
|2003        |955.0  |30082.0  |92834.0...

除了0之外，这几乎是我所追求的图形，但我可以摆脱它。正如你所看到它仍然不是很易读，但这解决了我如何绘制图形的问题。

df.unstack(level=-1).plot(kind='bar', figsize = (10,10))

Final graph