在 python 中随时间绘制情绪分析

Question

我正在尝试 plot 随着时间的推移我的情绪分析结果。 该代码涉及来自论坛的评论。 我的代码示例如下所示：

Timestamp            Sentiment
2021-01-28 21:37:41  Positive
2021-01-28 21:32:10  Negative
2021-01-29 21:30:35  Positive
2021-01-29 21:28:57  Neutral
2021-01-29 21:26:56  Negative

我想 plot 一个折线图，其中只有 x 轴上时间戳的日期，然后是“情绪”列的值计数的单独行。 所以总共有 3 行，每个情绪（正面、负面和中性）各有一条，y 轴代表计数。 我想我需要以某种方式使用 groupby() 但我不知道如何。

Answer 1

我的解决方案有点复杂，您可能应该稍后增强图表以适应您想要的（如堆叠条）。

首先，让我们将您的 dataframe 时间戳分成日期。

import pandas as pd
import matplotlib.pyplot as plt
example = {'Timestamp':
          ['2021-01-28 21:37:41', '2021-01-28 21:32:10', '2021-01-29 21:30:35',
           '2021-01-29 21:28:57', '2021-01-29 21:26:56'],
           'Sentiment':
           ['Positive', 'Negative', 'Positive', 'Neutral', 'Negative']}
df = pd.DataFrame(example)
df['Timestamp'] = pd.to_datetime(df['Timestamp'])
df['Date'] = df['Timestamp'].dt.date

然后，让我们按日期分组，并计算唯一值。

grouped = df.groupby(by='Date')['Sentiment'].value_counts()

Output：

Date        Sentiment
2021-01-28  Negative     1
            Positive     1
2021-01-29  Negative     1
            Neutral      1
            Positive     1
Name: Sentiment, dtype: int64

这是一个多指标系列。 为了获得更好的格式，我们可以取消堆叠多索引。

unstacked = grouped.unstack(level=1)

然后，我们可以直接将 plot 放在 object 上，unstacked.plot.bar unstacked.plot.bar() 。 这就是结果。