在python中创建条件轴标签

import numpy as np

# 创建一个数组
arr = np.array([0, 1, 2, 3, 4, 5])

# 使用np.where()根据条件创建轴标签
labels = np.where(arr > 2, '大于2', '小于等于2')

print(labels)

['小于等于2' '小于等于2' '小于等于2' '大于2' '大于2' '大于2']

import pandas as pd

# 创建一个Series对象
s = pd.Series([0, 1, 2, 3, 4, 5])

# 创建映射关系
mapping = {0: '零', 1: '一', 2: '二', 3: '三', 4: '四', 5: '五'}

# 使用map()方法创建轴标签
labels = s.map(mapping)

print(labels)

0    零
1    一
2    二
3    三
4    四
5    五
dtype: object