java-codingbat/python/String-1/first_two.py at master · sum-it/java-codingbat · GitHub

10 lines (9 loc) · 398 Bytes

#Given a string, return the string made of its first two chars, so the String "Hello" yields "He". If the string is shorter than length 2, return whatever there is, so "X" yields "X", and the empty string "" yields the empty string "".
#first_two('Hello') → 'He'
#first_two('abcdefg') → 'ab'
#first_two('ab') → 'ab'
def first_two(str):
  if (len(str)<=2):
    return str
  return str[0:2]