计算角色在电影剧本中所说的单词

时间:2018-04-17 14:55:57

标签: python python-3.x text count movie

我已经设法在一些帮助下发现了口语。 现在,我正在寻找被选中的人所说的文字。 所以我可以输入MIA并获取她在电影中所说的每一个字 像这样:

name = input("Enter name:")
wordsspoken(script, name)
name1 = input("Enter another name:")
wordsspoken(script, name1)

所以我可以计算之后的单词。

这就是电影剧本的样子

An awkward beat. They pass a wooden SALOON -- where a WESTERN
 is being shot. Extras in COWBOY costumes drink coffee on the
 steps.
                     Revision                        25.


                   MIA (CONT'D)
      I love this stuff. Makes coming to work
      easier.

                   SEBASTIAN
      I know what you mean. I get breakfast
      five miles out of the way just to sit
      outside a jazz club.

                   MIA
      Oh yeah?

                   SEBASTIAN
      It was called Van Beek. The swing bands
      played there. Count Basie. Chick Webb.
             (then,)
      It's a samba-tapas place now.

                   MIA
      A what?

                   SEBASTIAN
      Samba-tapas. It's... Exactly. The joke's on
      history.

4 个答案:

答案 0 :(得分:2)

我会首先向用户询问脚本中的所有名称。然后问他们想要哪个名字。我会逐字逐句搜索文本,直到找到想要的名字并将下面的单词复制到变量中,直到我找到一个与脚本中的其他人匹配的名称。现在人们可以说出另一个角色的名字,但是如果你假设人们说话的标题要么全部大写,要么单行,那么文本应该很容易过滤。

for word in script:
    if word == speaker and word.isupper(): # you may want to check that this is on its own line as well.
        recording = True
    elif word in character_names and word.isupper():  # you may want to check that this is on its own line as well.
        recording = False

    if recording:
        spoken_text += word + " "

答案 1 :(得分:2)

我将概述如何生成一个dict,它可以为您提供所有发言者所说的单词数量,以及一个近似于您现有实现的单词。

一般用途

如果我们将一个单词定义为沿着' ' (空间)...

import re

speaker = '' # current speaker
words = 0 # number of words on line
word_count = {} # dict of speakers and the number of words they speak

for line in script.split('\n'):
    if re.match('^[ ]{19}[^ ]{1,}.*', line): # name of speaker
            speaker = line.split(' (')[0][19:]
    if re.match('^[ ]{6}[^ ]{1,}.*', line): # dialogue line
            words = len(line.split())
            if speaker in word_count:
                 word_count[speaker] += words
            else:
                 word_count[speaker] = words

如果John Doe说55个单词,则生成格式为{'JOHN DOE':55}的dict。

示例输出:

>>> word_count['MIA']

13

您的实施

以上是上述程序的一个版本,它近似于您的实现。

import re

def wordsspoken(script,name):
    word_count = 0
    for line in script.split('\n'):
        if re.match('^[ ]{19}[^ ]{1,}.*', line): # name of speaker
            speaker = line.split(' (')[0][19:]
        if re.match('^[ ]{6}[^ ]{1,}.*', line): # dialogue line
            if speaker == name:
                word_count += len(line.split())
    print(word_count)

def main():
    name = input("Enter name:")
    wordsspoken(script, name)
    name1 = input("Enter another name:")
    wordsspoken(script, name1)

答案 2 :(得分:1)

如果你想通过脚本只有一次传递计算你的计数(我想这可能很长),你可以只跟踪说话的角色;设置就像一个小状态机:

import re
from collections import Counter, defaultdict

words_spoken = defaultdict(Counter)
currently_speaking = 'Narrator'

for line in SCRIPT.split('\n'):
    name = line.replace('(CONT\'D)', '').strip()
    if re.match('^[A-Z]+$', name):
        currently_speaking = name
    else:
        words_spoken[currently_speaking].update(line.split())

您可以使用更复杂的正则表达式来检测发言者何时发生变化,但这应该可以解决问题。

demo

答案 3 :(得分:1)

上面有一些好主意。以下内容在Python 2.x和3.x中应该可以正常工作:

import codecs
from collections import defaultdict

speaker_words = defaultdict(str)

with codecs.open('script.txt', 'r', 'utf8') as f:
  speaker = ''
  for line in f.read().split('\n'):
    # skip empty lines
    if not line.split():
      continue

    # speakers have their names in all uppercase
    first_word = line.split()[0]
    if (len(first_word) > 1) and all([char.isupper() for char in first_word]):
      # remove the (CONT'D) from a speaker string
      speaker = line.split('(')[0].strip()

    # check if this is a dialogue line
    elif len(line) - len(line.lstrip()) == 6:
      speaker_words[speaker] += line.strip() + ' '

# get a Python-version-agnostic input
try:
  prompt = raw_input
except:
  prompt = input

speaker = prompt('Enter name: ').strip().upper()
print(speaker_words[speaker])

示例输出:

Enter name: sebastian
I know what you mean. I get breakfast five miles out of the way just to sit outside a jazz club. It was called Van Beek. The swing bands played there. Count Basie. Chick Webb. It's a samba-tapas place now. Samba-tapas. It's... Exactly. The joke's on history.