假设shape是数据框中的一列:
shape color
circle rectangle orange
square triangle
rombus
square oval black
triangle circle
rectangle oval white
triangle
我想在形状列中计算数据框中有多少圆形,矩形,椭圆形,三角形.
输出应为:
circle 2
rectangle 2
triangle 3
oval 1
解决方法:
采用:
L = ['circle','rectangle','oval','triangle']
s = df['shape'].astype(str).str.split(expand=True).stack()
df = s[s.isin(L)].value_counts().reindex(L, fill_value=0).reset_index()
df.columns = ['vals','counts']
print (df)
vals counts
0 circle 2
1 rectangle 2
2 oval 2
3 triangle 3
说明:
>空格开头的split
(默认分隔符)和单词系列的stack
>按列表中的值按isin
筛选
>用于计数使用value_counts
>如有必要,更改顺序或将缺失值添加为0,然后添加reindex
>对于来自系列的DataFrame,请添加reset_index