python 读取srt文件并把字幕添加到视频中-CFANZ编程社区

Python 读取srt文件并把字幕添加到视频中

在视频制作过程中，添加字幕是一项非常常见的任务。Python 提供了强大的库来处理视频和字幕文件，使我们能够轻松地将字幕添加到视频中。本文将介绍如何使用 Python 读取 srt 文件并将字幕添加到视频中的方法。

1. 安装依赖库

在开始之前，我们需要安装一些 Python 库来处理视频和字幕文件。我们将使用以下两个库：

moviepy：处理视频文件的库。
pysrt：处理字幕文件的库。

可以使用以下命令在终端中安装这些库：

pip install moviepy pysrt

2. 读取srt文件

首先，我们需要加载 srt 文件并将其解析为易于处理的格式。pysrt 库提供了一个简单的方法来做到这一点。以下是一个示例代码，演示了如何加载和解析 srt 文件：

import pysrt

def load_srt_file(file_path):
    subs = pysrt.open(file_path)
    
    for sub in subs:
        print(f"Start: {sub.start}, End: {sub.end}")
        print(f"Text: {sub.text}")
        print("--------")

srt_file = "subtitle.srt"
load_srt_file(srt_file)

在这个示例中，我们定义了一个 load_srt_file 函数，它接受一个 srt 文件的路径作为参数。pysrt.open 函数用于加载 srt 文件并返回一个 SubRipFile 对象。我们可以使用这个对象来遍历所有的字幕。在这个例子中，我们简单地打印了每个字幕的开始时间、结束时间和文本内容。

3. 添加字幕到视频

现在，我们已经加载并解析了 srt 文件，下一步是将其添加到视频中。为了完成这个任务，我们将使用 moviepy 库。以下是一个示例代码，演示了如何将字幕添加到视频中：

from moviepy.editor import VideoFileClip, TextClip
from moviepy.video.tools.subtitles import SubtitlesClip

def add_subtitles_to_video(video_file, srt_file, output_file):
    video = VideoFileClip(video_file)
    subs = pysrt.open(srt_file)

    def subtitle_text(t):
        index = subs.index_nearest(t)
        return subs[index].text if index else ""

    sub_clip = SubtitlesClip(subs, subtitle_text)
    sub_clip = sub_clip.set_duration(video.duration)

    result = CompositeVideoClip([video, sub_clip.set_position(("center", "bottom"))])
    result.write_videofile(output_file, codec="libx264", audio_codec="aac")

video_file = "video.mp4"
srt_file = "subtitle.srt"
output_file = "output.mp4"

add_subtitles_to_video(video_file, srt_file, output_file)

在这个示例中，我们定义了一个 add_subtitles_to_video 函数，它接受视频文件路径、srt 文件路径和输出文件路径作为参数。首先，我们使用 VideoFileClip 函数加载视频文件。然后，我们加载并解析了 srt 文件，就像在前面的步骤中所做的那样。

然后，我们定义了一个 subtitle_text 函数来根据视频的当前时间返回相应的字幕文本。这个函数使用 subs.index_nearest 方法来找到最接近给定时间的字幕，并返回其文本内容。

接下来，我们使用 SubtitlesClip 函数创建一个字幕剪辑。我们将其持续时间设置为视频的持续时间，并将其位置设置为视频的底部中心。

最后，我们使用 CompositeVideoClip 函数将视频和字幕剪辑合并在一起。然后，我们使用 write_videofile 方法将最终结果写入输出文件。