1873. 计算特殊奖金
2023年8月12日20:24:22
简单
SQL Schema
Pandas Schema
表: Employees
+-------------+---------+
| 列名 | 类型 |
+-------------+---------+
| employee_id | int |
| name | varchar |
| salary | int |
+-------------+---------+
employee_id 是这个表的主键(具有唯一值的列)。
此表的每一行给出了雇员id ，名字和薪水。
编写解决方案，计算每个雇员的奖金。如果一个雇员的 id 是 奇数 并且他的名字不是以 'M' 开头，那么他的奖金是他工资的 100% ，否则奖金为 0 。
返回的结果按照 employee_id 排序。
返回结果格式如下面的例子所示。
示例 1:
输入：
Employees 表:
+-------------+---------+--------+
| employee_id | name | salary |
+-------------+---------+--------+
| 2 | Meir | 3000 |
| 3 | Michael | 3800 |
| 7 | Addilyn | 7400 |
| 8 | Juan | 6100 |
| 9 | Kannon | 7700 |
+-------------+---------+--------+
输出：
+-------------+-------+
| employee_id | bonus |
+-------------+-------+
| 2 | 0 |
| 3 | 0 |
| 7 | 7400 |
| 8 | 0 |
| 9 | 7700 |
+-------------+-------+
解释：
因为雇员id是偶数，所以雇员id 是2和8的两个雇员得到的奖金是0。
雇员id为3的因为他的名字以'M'开头，所以，奖金是0。
其他的雇员得到了百分之百的奖金。
通过次数
82.9K
提交次数
132.6K
通过率
62.5%
答案
import pandas as pd
def calculate_special_bonus(employees: pd.DataFrame) -> pd.DataFrame:
# employees["bonus"] = employees.apply(lambda x :x["salary"] if x["employee_id"]%2 and x["name"][0]!="M" else 0,axis=1)
# employees = employees.sort_values(by=["employee_id"])
# return employees[["employee_id","bonus"]]
employees['bonus'] = employees[ (employees["employee_id"] % 2 != 0) & (employees["name"].str[0] != "M") ]['salary']
return employees[['employee_id','bonus']].fillna(0).sort_values(by="employee_id",ascending=True)