如何从pandas DataSeries中提取唯一的排列?

使用Pandas DataSeries在Jupyter中工作我有一个包含如下行的数据集:

color: white
engineType: diesel
make: Ford
manufacturingYear: 2004
accidentCount: 123

我需要做的是绘制颜色/ engineType / make的所有排列的制造年份(x轴)的事故计数图表(y轴).任何想法如何处理这个?

为了加快速度,我有了这个初始设置:

import numpy as np
import pandas as pd
from pandas import DataFrame, Series
import random


colors = ['white', 'black','silver']
engineTypes = ['diesel', 'petrol']
makes = ['ford', 'mazda', 'subaru']
years = range(2000,2005)

rowCount = 100

def randomEl(data):
    rand_items = [data[random.randrange(len(data))] for item in range(rowCount)]
    return rand_items


df = DataFrame({
    'color': Series(randomEl(colors)),
    'engineType': Series(randomEl(engineTypes)),
    'make': Series(randomEl(makes)),
    'year': Series(randomEl(years)),
    'accidents': Series([int(1000*random.random()) for i in range(rowCount)])
})
最佳答案
您可以使用groupby()通过独特颜色,engineType和组合来获取事故数量:

accident_counts = df.groupby(['color', 'engineType', 'make'])['accidents'].sum()

Matplotlib是绘制结果的一种方法:

import matplotlib.pyplot as plt
accident_counts.plot(kind='bar')
plt.show()

转载注明原文:如何从pandas DataSeries中提取唯一的排列? - 代码日志