LeetCode 434 Number of Segments in a String

class Solution(object):
    # https://leetcode.com/problems/number-of-segments-in-a-string/solution/
    # def countSegments(self, s):
    #     """
    #     :type s: str
    #     :rtype: int
    #     """
    #     return len(s.split())

    def countSegments(self, s):
         segment_count = 0
        for i in range(len(s)):
            if (i == 0 or s[i-1] == ' ') and s[i] != ' ':
                segment_count += 1

        return segment_count