基于列对数据框进行分组

凯尔蒂

我是数据科学的新手，感谢您的帮助。我的问题是关于基于列对数据框进行分组，以便根据每个主题状态绘制条形图

我的 csv 文件是这样的

Name,Maths,Science,English,sports
S1,Pass,Fail,Pass,Pass
S2,Pass,Pass,NA,Pass
S3,Pass,Fail,Pass,Pass
S4,Pass,Pass,Pass,NA
S5,Pass,Fail,Pass,NA

预期o/p：

Subject,Status,Count
Maths,Pass,5
Science,Pass,2
Science,Fail,3
English,Pass,4
English,NA,1
Sports,Pass,3
Sports,NA,2

隆皮

您可以使用 Pandas 执行此操作，与问题中的输出格式不完全相同，但绝对具有相同的信息：

import pandas as pd

# reading csv
df = pd.read_csv("input.csv")

# turning columns into rows
melt_df = pd.melt(df, id_vars=['Name'], value_vars=['Maths', 'Science', "English", "sports"], var_name="Subject", value_name="Status")

# filling NaN values, otherwise the below groupby will ignore them.
melt_df = melt_df.fillna("Unknown")

# counting per group of subject and status.
result_df = melt_df.groupby(["Subject", "Status"]).size().reset_index(name="Count")

然后你会得到以下结果：

   Subject   Status  Count
0  English     Pass      4
1  English  Unknown      1
2    Maths     Pass      5
3  Science     Fail      3
4  Science     Pass      2
5   sports     Pass      3
6   sports  Unknown      2

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。