Распечатать элемент списка и его X последующие

Question

Список ввода:

a = ['Dog', 'Cat', 'Bird', 'Horse']

Мой код:

For i in range(len(a)):
    word = a[i]
    if word == 'Cat'
       b = open('file.txt','w')
       b.write(a[i])

Пока код повторяется и записывает Cat в новый файл, но я хотел бы, чтобы моя программа писала Catа также последующие строки, которые в этом случае будут Bird и Horse.

Могу ли я что-нибудь сделать?т.е.: b.write(a[i:]) или b.write(a[i:2])

Редактировать: Решено с использованием следующего:

b.write('\n'.join(a[i:i+X])), где X - последующие элементы.

Я все еще открыт для лучших ответов.Благодаря.

Answer 1

вы можете пропустить цикл for и просто сделать:

with open('f', 'w') as f:
    f.write('\n'.join(a[a.index('Cat'):]))

Answer 2

Вы можете сделать:

b.write('\n'.join(a[i:]))

, если вы хотите, чтобы символ новой строки был разделен между символами, если вам нужен пробел, просто используйте ' '.join(a[i:]) например