python - 创建一个新的数据框，统计每个用户的正面和负面推文

Question

我有以下数据框:

它包含 user_ids、推文、位置以及推文的负面和正面分类。

我想创建一个按用户 ID 分组的新数据框，因为每个用户在数据框中都有多个推文。数据框应包含以下列:

1. 用户 ID
2. 该 user_id 发布的负面推文数量
3. 该 user_id 发布的正面推文数量
4. 用户所在位置

所需的示例输出

user_id             positive_tweets   negative_tweets    Location
418                 1                    0                   CA
521                 1                    0                   CA
997                 0                    1                   LA
1135                1                    0                   LA

这个代码是 BlackFox 先生针对我之前没有问正确的问题建议的。

df.groupby(['user_id','classification'])['user_id'].count()

但是，它与所需的输出不匹配。

谢谢

Answer 1

我希望这就是您正在寻找的。

df.groupby(['user_id', 'Location']).apply(lambda x: pd.Series(dict(
positive_tweets=(x.classification == 'positive').sum(),
negative_tweets=(x.classification == 'negative').sum(),
)))