如何在Python groupby中分离出现的事件

Question

因此，我正在尝试创建一个简单的饼图，以形象化政治情绪（刮擦Twitter）。我分为三类：negative，neutral和positive。 到目前为止，我有：

import seaborn as sns
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib import style
raw_data = 'xxxxxxxx/trud.csv'
df = pd.read_csv(raw_data, index_col = 0)
test = df.groupby('sent').sent.count(). # 'sent' is a column 
print(test)

输出：

negative 178
neutral 359
positive 263

我可以分开每个输出并给它一个变量，以便制作饼图/条形图吗？ neg = negative，neu = neutral等。 谢谢！

Answer 1

您可以按索引选择Series的值：

test = pd.Series([178, 359, 263], index=['negative','neutral','positive'])

neg = test['negative']
neu = test['neutral']
pos = test['positive']

但是对于绘图不是必需的-使用Series.plot.pie：

test.plot.pie(figsize=(3, 3))

或Series.plot.bar：

test.plot.bar(colors=['r','b','y'])