Python如何获取"视频文件"的大小和时长?
今天微信群有一个朋友问了下面这个问题,虽然很简单,但是觉得有必要和大家做一个知识分享。
这个问题一共有两个需求:
① 获取视频的文件大小;
② 获取视频的时长;
基于上述两个问题,我来教大家如何将获取到的信息存储到Excel中。
1. 前置知识
这里一共需要用到3个Python库,分别是os、pandas、moviepy。
① os:用于获取某个目录下的文件和文件大小;
② pandas:用于将提取到的信息,保存到Excel中;
③ moviepy:用于获取视频文件中的时长;
注意:pandas和moviepy属于第三方库,因此需要我们提前使用如下命令进行安装。
pip install pandas
pip install moviepy
2. 完整代码
① 导入所有库
import os
import pandas as pd
from moviepy.editor import VideoFileClip
② 获取指定目录下符合条件的文件
useful_dir = []
for i in os.listdir():
if i.endswith("wmv"):
useful_dir.append(i)
useful_dir
结果如下:
③ 自定义时间转化函数
def time_convert(seconds):
M,H = 60,3600
if seconds < M:
return f'00:00:0{seconds}' if seconds < 10 else f'00:00:{str(seconds)}'
elif seconds < H:
_M = int(seconds//M)
_S = int(seconds%M)
return f'00:{f"0{_M}" if _M < 10 else str(_M)}:{f"0{_S}" if _S < 10 else str(_S)}'
else:
_H = int(seconds//H)
_M = int(seconds%H//M)
_S = int(seconds%H%M)
return f'{f"0{_H}" if _H < 10 else str(_H)}:{f"0{_M}" if _M < 10 else str(_M)}:{f"0{_S}" if _S < 10 else str(_S)}'
由于下面介绍的VideoFileClip()
方法,获取时长的单位是秒。因此,需将秒换成合适的"时分秒"格式。若时间超过一分钟,换算成"分钟:秒",若超过一小时,就换算成"小时:分钟:秒"格式。
④ 获取文件大小和时长
x = []
y = []
for i in useful_dir:
dir_size = str(round(os.path.getsize(i)/1024/1024,1)) + "M"
clip = VideoFileClip(i)
dir_time = time_convert(clip.duration)
x.append(dir_size)
y.append(dir_time)
df = pd.DataFrame({"文件大小":x,"文件时长":y})
df
结果如下:
⑤ 将得到的数据存储到Excel中
df.to_excel("info.xlsx",index=False)
最终效果如图所示:
END
推荐阅读
吴恩达deeplearining.ai的经典总结资料
Ps:从小程序直接获取下载