Python：使用 Pandas 进行分箱和可视化

Question

我对 python 还是很陌生。

所以我正在尝试为我的数据框创建一个年龄间隔列

df['age_interval'] = pd.cut(x=df['Age'], bins=[18, 22, 27, 32, 37, 42, 47, 52, 57, 60], include_lowest=True)

我添加了我的图表：

问题：在可视化中，[18-22] bin 显示为 [17.99-22]

我想要什么：我希望它显示 18-22。

下面是剧情代码：

plt.figure(figsize=(15,8))
dist = sns.barplot(x=ibm_ages.index, y=ibm_ages.values, color='blue')
dist.set_title('IBM Age Distribution', fontsize = 24)
dist.set_xlabel('Age Range', fontsize=18)
dist.set_ylabel('Total Count', fontsize=18)

sizes=[]
for p in dist.patches:
    height = p.get_height()
    sizes.append(height)
    dist.text(p.get_x()+p.get_width()/2.,
            height + 5,
            '{:1.2f}%'.format(height/total*100),
            ha="center", fontsize= 8) 

plt.tight_layout(h_pad=3)
plt.show()

谢谢

Answer 1

那是因为它是一个 float64 类型，你想要一个整数试试：

import numpy as np
df['age_interval'] = pd.cut(x=df['Age'].astype('Int64'), bins=[18, 22, 27, 32, 37, 42, 47, 52, 57, 60], include_lowest=True)

您可以在任何时候将 float64 转换为 Int64 时使用 .astype('Int64')