Python3 使用 pymediainfo 统计视频信息并写入 Excel
以下是一个使用 pymediainfo 库统计视频 info 并将结果写入 excel 的 Python3 脚本示例:
import os
import argparse
import pandas as pd
from pymediainfo import MediaInfo
def get_video_info(filename):
'获取视频信息'
media_info = MediaInfo.parse(filename)
for track in media_info.tracks:
if track.track_type == 'Video':
return {
'codec': track.codec,
'width': track.width,
'height': track.height,
'duration': track.duration / 1000.0,
'frame_rate': track.frame_rate,
'bit_rate': track.bit_rate / 1000.0
}
def write_to_excel(video_files, output_file):
'将视频信息写入excel'
rows = []
for video_file in video_files:
if os.path.isfile(video_file):
video_info = get_video_info(video_file)
if video_info:
rows.append(video_info)
if rows:
df = pd.DataFrame(rows)
df.to_excel(output_file, index=False)
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('-i', '--input', nargs='+', required=True, help='输入视频文件路径,可以指定多个')
parser.add_argument('-o', '--output', required=True, help='输出excel文件路径')
args = parser.parse_args()
write_to_excel(args.input, args.output)
使用示例:
$ python3 video_info_to_excel.py -i video.mp4 -o output.xlsx
其中,-i 参数指定输入的视频文件路径,可以指定多个,用空格分隔;-o 参数指定输出的 excel 文件路径。脚本将会读取指定的视频文件,获取视频信息,将结果写入指定的 excel 文件中。
原文地址: https://www.cveoy.top/t/topic/mz04 著作权归作者所有。请勿转载和采集!