该题目来源于力扣:
题目要求:
编写一个解决方案,基于名为 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 pd
def createDataframe(student_data: List[List[int]]) -> pd.DataFrame:
DataFrame_New=pd.DataFrame(student_data,columns=['student_id','age'])
return DataFrame_New