본문 바로가기
나는 이렇게 학습한다/Algorithm & SQL

Find the capitals

by daco2020 2022. 5. 9.

Instructions

Write a function that takes a single string (word) as argument. The function must return an ordered list containing the indexes of all capital letters in the string.

Example

Test.assertSimilar( capitals('CodEWaRs'), [0,3,4,6] );

 

 

Solution:

1. Find the uppercase character in the string.
2. It returns the index of the corresponding character in an array.

 

def capitals(word):
    return [i for i, v in enumerate(word) if v.isupper()]