使用多个条件语句逐行向量化pandas df

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [6, 7, 8, 9, 10],
        'C': [11, 12, 13, 14, 15]}
df = pd.DataFrame(data)

# 定义一个函数，该函数将根据不同的条件返回不同的值
def custom_function(row):
    if row['A'] < 3:
        return 'Low'
    elif row['A'] < 5:
        return 'Medium'
    else:
        return 'High'

# 使用apply函数应用自定义函数到DataFrame的每一行，并创建一个新列来存储结果
df['Category'] = df.apply(custom_function, axis=1)

# 打印结果
print(df)

   A   B   C Category
0  1   6  11      Low
1  2   7  12      Low
2  3   8  13   Medium
3  4   9  14   Medium
4  5  10  15     High