[Study]/[Pandas]

[LeetCode - Pandas] (Easy) 2884. Modify Columns

잰잰' 2025. 3. 30. 21:58
DataFrame employees
+-------------+--------+
| Column Name | Type   |
+-------------+--------+
| name        | object |
| salary      | int    |
+-------------+--------+​

회사는 직원들에게 급여 인상을 제공하려고 합니다.

급여 열을 수정하여 각 급여를 2배로 곱하는 해결책을 작성하세요.

결과 형식은 다음 예시와 같습니다.

Example 1:
Input:
DataFrame employees
+---------+--------+
| name    | salary |
+---------+--------+
| Jack    | 19666  |
| Piper   | 74754  |
| Mia     | 62509  |
| Ulysses | 54866  |
+---------+--------+
Output:
+---------+--------+
| name    | salary |
+---------+--------+
| Jack    | 39332  |
| Piper   | 149508 |
| Mia     | 125018 |
| Ulysses | 109732 |
+---------+--------+

설명: 모든 급여가 두 배로 증가했습니다.

 

✏️ 풀이

import pandas as pd

def modifySalaryColumn(employees: pd.DataFrame) -> pd.DataFrame:
    employees['salary'] = employees['salary'] * 2

    return employees