[LeetCode - Pandas] (Easy) 2877. Create a DataFrame from List

주어진 2D 리스트인 student_data로부터 DataFrame을 생성하는 솔루션을 작성하세요. 이 2D 리스트는 몇 명의 학생들의 ID와 나이를 포함하고 있습니다.

DataFrame은 두 개의 열, student_id와 age를 가져야 하며, 원래 2D 리스트와 같은 순서로 작성되어야 합니다.

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

Example 1:

Input:
student_data:
[
  [1, 15],
  [2, 11],
  [3, 11],
  [4, 20]
]
Output:
+------------+-----+
| student_id | age |
+------------+-----+
| 1          | 15  |
| 2          | 11  |
| 3          | 11  |
| 4          | 20  |
+------------+-----+

설명 : student_data 위에 DataFrame이 생성되었으며, 두 개의 열은 student_id와 age라는 이름을 가지고 있습니다.

 

✏️ 풀이

import pandas as pd

def createDataframe(student_data: List[List[int]]) -> pd.DataFrame:
    return pd.DataFrame(student_data, columns=['student_id', 'age'])