如何根据pandas中的条件划分值并放置在下一列中

import pandas as pd
import numpy as np

# 创建一个示例DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [10, 20, 30, 40, 50]}
df = pd.DataFrame(data)

# 使用np.where()函数根据条件划分值并放置在下一列中
df['C'] = np.where(df['A'] > 3, '大于3', '小于等于3')

print(df)

   A   B    C
0  1  10  小于等于3
1  2  20  小于等于3
2  3  30  小于等于3
3  4  40    大于3
4  5  50    大于3

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [10, 20, 30, 40, 50]}
df = pd.DataFrame(data)

# 使用DataFrame.loc方法根据条件划分值并放置在下一列中
df.loc[df['A'] > 3, 'C'] = '大于3'
df.loc[df['A'] <= 3, 'C'] = '小于等于3'

print(df)

   A   B    C
0  1  10  小于等于3
1  2  20  小于等于3
2  3  30  小于等于3
3  4  40    大于3
4  5  50    大于3