Counting multiple letter groups in a string

后端 未结 2 1533
北荒
北荒 2020-12-21 19:28

I\'ve been trying to adapt my python function to count groups of letters instead of single letters and I\'m having a bit of trouble. Here\'s the code I have to count individ

2条回答
  •  小蘑菇
    小蘑菇 (楼主)
    2020-12-21 20:17

    This can be done pretty quickly using collections.Counter.

    from collections import Counter
    
    s = "CTAACAAC"
    
    def chunk_string(s, n):
        return [s[i:i+n] for i in range(len(s)-n+1)]
    
    counter = Counter(chunk_string(s, 3))
    # Counter({'AAC': 2, 'ACA': 1, 'CAA': 1, 'CTA': 1, 'TAA': 1})
    

    Edit: To elaborate on chunk_string:

    It takes a string s and a chunk size n as arguments. Each s[i:i+n] is a slice of the string that is n characters long. The loop iterates over the valid indices where the string can be sliced (0 to len(s)-n). All of these slices are then grouped in a list comprehension. An equivalent method is:

    def chunk_string(s, n):
        chunks = []
        last_index = len(s) - n
        for i in range(0, last_index + 1):
            chunks.append(s[i:i+n])
        return chunks
    

提交回复
热议问题