python统计单词个数

输入一个英文句子，以回车符结束，单词间以空格分隔，标点符号后跟至少一个空格，统计并输出单词的个数（仅统计单词，数字按单词计，不计算标点符号，重复单词出现几次就统计几次）。

测试输入： Life is short, you need Python!

预期输出： 6

 

import string
s = input()
count = 0 if s[0] == ' ' else 1
for i in range(len(s)):
    if s[i] == ' ' and s[i + 1] in string.ascii_letters+string.digits:
        count += 1
print(count)