如何在pandas中对列进行分类并添加新列

import pandas as pd

# 创建一个DataFrame对象
df = pd.DataFrame({'column_name': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]})

# 对列进行分类并添加新列
df['category'] = pd.cut(df['column_name'], bins=[0, 5, 10], labels=['low', 'high'])

# 打印结果
print(df)

   column_name category
0            1      low
1            2      low
2            3      low
3            4      low
4            5      low
5            6     high
6            7     high
7            8     high
8            9     high
9           10     high