如何通过将pandas DataFrame从数据框中的另一列创建pandas DataFrame？

Question

我有以下来源数据框架

国家	国家	富裕？
0	美国	是
1	印度	否
2	印度	是
3	我们	是
4	我们	否
是5	印度	US 6
6	US	7
印度	否	我需要将

其转换另一个数据框架，以绘制以下类似的条形图，以便轻松访问数据

每个国家/地区的经济状态图表

要创建的数据框架如下。

国家	贫穷的	贫穷
美国	3	1
印度	1	3

我是熊猫和探索性数据科学的新手。请在这里帮忙

Answer 1

您可以尝试pivot_table

df['Is Rich?'] = df['Is Rich?'].replace({'Yes': 'Rich', 'No': 'Poor'})
out = df.pivot_table(index='Country', columns='Is Rich?', values='Person', aggfunc='count')

print(out)

Is Rich?  Poor  Rich
Country
India        3     1
US           1     3

Answer 2

您可以这样做：

converted = df.assign(Rich=df['Is Rich?'].eq('Yes')).eval('Poor = ~Rich').groupby('Country').agg({'Rich': 'sum', 'Poor': 'sum'})

print(converted)

         Rich  Poor
Country            
India       1     3
US          3     1

但是，如果要将其绘制为barplot，则以下格式可能与绘图库（如 seaborn ：

plot_df = converted.reset_index().melt(id_vars='Country', value_name='No. of people', var_name='Status')
print(plot_df)

  Country Status  No. of people
0   India   Rich              1
1      US   Rich              3
2   India   Poor              3
3      US   Poor              1

然后，使用seaborn：

import seaborn as sns

sns.barplot(x='Country', hue='Status', y='No. of people', data=plot_df)

结果图：