Python – Extract K length substrings

The task is to extract all possible substrings of a specific length, k. This problem involves identifying and retrieving those substrings in an efficient way. Let’s explore various methods to extract substrings of length k from a given string in Python

Using List Comprehension

List comprehension is the most efficient and concise way to extract substrings. It iterates through the string and collects all substrings of length k in a single line.

s = "geeksforgeeks"
k = 4

# Extracting k-length substrings using list comprehension
sub = [s[i:i+k] for i in range(len(s) - k + 1)]
print(sub)  

['geek', 'eeks', 'eksf', 'ksfo', 'sfor', 'forg', 'orge', 'rgee', 'geek', 'eeks']

Explanation:

• List comprehension iterates over the string and slices it to get all substrings of length k.
• The range ensures we only extract valid substrings within the bounds of the string.

Let’s explore some more methods and see how we can extract K length substrings from a given string.

Using a for loop

By using a simple for loop, we can extract the substrings by iterating through the string and slicing it at each step. This method is very intuitive.

s = "geeksforgeeks"
k = 4

# Initialize an empty list to store substrings
sub = []

# Loop through the string and extract k-length substrings
for i in range(len(s) - k + 1):
    sub.append(s[i:i+k])
print(sub) 

['geek', 'eeks', 'eksf', 'ksfo', 'sfor', 'forg', 'orge', 'rgee', 'geek', 'eeks']

Explanation:

• A for loop iterates through the string, slicing substrings of length k.
• Each substring is appended to the substrings list.

Using zip()

zip() function can also be used for this task, we can use it to group characters of the string into k-length substrings.

s = "geeksforgeeks"
k = 4

# Use zip to extract k-length substrings
substrings = [''.join(s[i:i+k]) for i in range(len(s) - k + 1)]
print(substrings) 

['geek', 'eeks', 'eksf', 'ksfo', 'sfor', 'forg', 'orge', 'rgee', 'geek', 'eeks']

Explanation:

• The zip() function groups characters in chunks of size k.
• A list comprehension is used to extract and join these chunks into substrings.

Using re.findall()

re.findall() function can be used to extract substrings based on regular expressions.

import re
s = "geeksforgeeks"
k = 4

# Use re.findall to extract k-length substrings
subs = re.findall(r'(?=(\w{4}))', s)
print(subs) 

['geek', 'eeks', 'eksf', 'ksfo', 'sfor', 'forg', 'orge', 'rgee', 'geek', 'eeks']

Explanation:

• The regular expression (?=(\w{4})) matches overlapping substrings of length 4.
• This method uses regular expressions, which may be overkill for simple tasks.

manjeet_04

Follow

Improve

Article Tags :

• Python
• Python Programs
• Python string-programs

Practice Tags :

• python