前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Python读取Excel文件统计演员参演电影

Python读取Excel文件统计演员参演电影

作者头像
Python小屋屋主
发布2018-04-16 17:25:44
2.4K0
发布2018-04-16 17:25:44
举报
文章被收录于专栏:Python小屋

假设有个Excel 2007文件“电影导演演员.xlsx”,其中有三列分别为电影名称、导演和演员列表(同一个电影可能会有多个演员,每个演员姓名之间使用逗号分隔),现在要求统计每个演员的参演电影分别是哪些。

下面的代码使用到了扩展库openpyxl来读取xlsx文件,需要使用pip进行安装。

'''根据电影、导演、演员清单,统计每个演员的参演电影'''

import openpyxl

from openpyxl import Workbook

def getActors(filename):

actors = dict()

# 打开xlsx文件,并获取第一个worksheet

wb = openpyxl.load_workbook(filename)

ws = wb.worksheets[0]

# 遍历Excel文件中的所有行

for index, row in enumerate(ws.rows):

# 绕过第一行的表头

if index == 0:

continue

# 获取电影名称和演员列表

filmName, actor = row[0].value, row[2].value.split(',')

# 遍历该电影的所有演员,统计参演电影

for a in actor:

actors[a] = actors.get(a, set())

actors[a].add(filmName)

return actors

actors = getActors('电影导演演员.xlsx')

actors = sorted(actors.items(), key=lambda x:int(x[0][2:]))

for item in actors:

print(item)

代码执行结果为:

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2017-04-05,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 Python小屋 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档