pandas模块创建DataFrame对象df_data记录的是所有电影的属性信息,其中此对象的index属性为电影的名字(films列表)和电影的年代,属性columns 分别为所有特征(列表all attrs), 最后对此对象添加一列(列名字为“评分”)。电影名字:肖申克的救赎,霸王别姬。电影属性:犯罪 剧情 爱情 同性。肖申克的救赎 1 0 0 0。霸王别姬 0 1 1 1。评分9.7 9.6。
时间: 2023-02-13 11:00:40 浏览: 66
pandas创建DataFrame的7种方法小结
您可以使用以下代码来创建 DataFrame 对象并添加数据:
```
import pandas as pd
# 创建电影名字和年代的索引
films = [("肖申克的救赎", 1994), ("霸王别姬", 1993)]
index = pd.MultiIndex.from_tuples(films, names=['电影名字', '年代'])
# 创建电影属性的列名
all_attrs = ['犯罪', '剧情', '爱情', '同性']
# 创建电影属性的数据
data = [
[1, 0, 0, 0],
[0, 1, 1, 1]
]
# 创建 DataFrame 对象
df_data = pd.DataFrame(data, index=index, columns=all_attrs)
# 添加评分列
df_data['评分'] = [9.7, 9.6]
print(df_data)
```
输出结果:
```
犯罪 剧情 爱情 同性 评分
电影名字 年代
肖申克的救赎 1994 1 0 0 0 9.7
霸王别姬 1993 0 1 1 1 9.6
```
阅读全文