[每日一练]从表中创建DataFrame
该题目来源于力扣:
2877. 从表中创建 DataFrame - 力扣(LeetCode)https://leetcode.cn/problems/create-a-dataframe-from-list/description/?lang=pythondata
题目要求:
编写一个解决方案,基于名为 student_data 的二维列表 创建 一个 DataFrame 。这个二维列表包含一些学生的 ID 和年龄信息。DataFrame 应该有两列, student_id 和 age,并且与原始二维列表的顺序相同。返回结果格式如下示例所示。示例 1:输入:
student_data:
[[1, 15],[2, 11],[3, 11],[4, 20]
]
输出:
+------------+-----+
| student_id | age |
+------------+-----+
| 1 | 15 |
| 2 | 11 |
| 3 | 11 |
| 4 | 20 |
+------------+-----+
解释:
基于 student_data 创建了一个 DataFrame,包含 student_id 和 age 两列。
思路流程:
该题目要求我们可以通过pandas将二维数据列表转换为一个具有二维数据结构的表格。
特殊代码:pd.Dataframe(data,index,columns,dtype,copy)
- 参数1,data:传入的字典/列表/二维数据集
- 参数2,index:指定的行索引
- 参数3,columns:指定的列索引
- 参数4,dtype:定义的数据类型
- 参数5,copy:TRUE表示原始数据和新 DataFrame 之间不会共享数据,修改 DataFrame 不会影响原始数据。如果为 False,则 DataFrame 可能会使用传入数据的引用。这意味着对 DataFrame 的修改可能会影响原始数据。
由此可见,该题目的要求很简答,我们只需要使用参数3,columns设定列索引即可。
代码实现:
import pandas as pddef createDataframe(student_data: List[List[int]]) -> pd.DataFrame:DataFrame_New=pd.DataFrame(student_data,columns=['student_id','age'])return DataFrame_New