如何根据python中其他列的信息/条件将字符串放入新列

import pandas as pd

# 创建一个示例数据框
data = {
    '姓名': ['张三', '李四', '王五', '赵六'],
    '成绩': [85, 92, 78, 65]
}
df = pd.DataFrame(data)

# 定义一个函数来根据成绩分配等级
def assign_grade(score):
    if score >= 90:
        return 'A'
    elif score >= 80:
        return 'B'
    elif score >= 70:
        return 'C'
    else:
        return 'D'

# 应用函数到成绩列，并创建新列“等级”
df['等级'] = df['成绩'].apply(assign_grade)

print(df)

   姓名  成绩 等级
0  张三  85  B
1  李四  92  A
2  王五  78  C
3  赵六  65  D

import numpy as np

# 使用numpy的条件赋值
df['等级'] = np.select(
    [df['成绩'] >= 90, df['成绩'] >= 80, df['成绩'] >= 70],
    ['A', 'B', 'C'],
    default='D'
)