选择要p的列

Question

我参加了一个在线课程，我被困在一个部分，试图根据列名绘图。这不是课程的一部分，所以没有太多的指导。你知道吗

#column year rename
gdp_df.rename(columns=lambda x: x + "_GDP", inplace=True)
selfemployed_df.rename(columns=lambda x: x + "_SE", inplace=True)
salaried_df.rename(columns=lambda x: x + "_S", inplace=True)

#country column rename
gdp_df.rename(columns={"GDP per capita_GDP":"country"},inplace=True)
selfemployed_df.rename(columns={"Total self-employed (%)_SE":"country"}, inplace=True)
salaried_df.rename(columns={"Total salaried employees (%)_S":"country"}, inplace=True)

#merge the dataframe
merge_df = salaried_df.merge(gdp_df.merge(selfemployed_df, left_on='country', right_on='country', how='inner'), left_on='country', right_on='country',how='inner')

现在我被困在按美国、国内生产总值或使用率作图的步骤上。如何按列名包含“\u S”或“\u GDP”或“\u SE”的位置进行绘图分组？我做错了吗？你知道吗

Answer 1

嗯，错了。。。你可以用很多方法做事。。。 但是，如果要将所有数据集合并为一个数据集，并询问自己如何根据这些后缀提取列的子集，可以执行以下操作：

S_cols = merge_df.columns[merge_df.columns.str.endswith('_S')]

那么

merge_df[S_cols]

将为您提供一个数据帧，其中只有以“\u S”结尾的列。你知道吗

一种简单的方法就是

merge_df[S_cols].plot()

你可以想象其他结局也是如此。你知道吗

我不明白的是：当您自己合并数据帧时，您已经有了分离的数据帧，即：gdp_df、selfemployed_df和salaried_df。有什么问题吗？你知道吗