pandas模块创建DataFrame对象df_data记录的是所有电影的属性信息,其中此对象的index属性为电影的名字(films列表了和电影的年代,属性columns 分别为所有特征(列表all attrs), 最后对此对象添加一列(列名字为“评分”)。电影名字:肖申克的救赎,霸王别姬。电影属性:犯罪 剧情 爱情 同性。肖申克的救赎 1 0 0 0。霸王别姬 0 1 1 1。评分9.7 9.6。
时间: 2023-02-06 12:12:05 浏览: 18
pandas 中dataframe对象增加行与 dataframe对象修改
下面是一种创建 DataFrame 对象的方法:
```
import pandas as pd
# 定义电影名字列表
films = ['肖申克的救赎', '霸王别姬']
# 定义电影属性列表
all_attrs = ['犯罪', '剧情', '爱情', '同性']
# 定义电影属性值列表
values = [[1, 0, 0, 0], [0, 1, 1, 1]]
# 创建 DataFrame 对象
df_data = pd.DataFrame(values, index=films, columns=all_attrs)
# 添加一列
df_data['评分'] = [9.7, 9.6]
print(df_data)
```
输出结果如下:
```
犯罪 剧情 爱情 同性 评分
肖申克的救赎 1 0 0 0 9.7
霸王别姬 0 1 1 1 9.6
```
阅读全文