Python program to find the character position of Kth word from a list of strings

# Python3 code to demonstrate working of
# Word Index for K position in Strings List
# Using enumerate() + list comprehension

# initializing list
test_list = ["geekforgeeks", "is", "best", "for", "geeks"]

# printing original list
print("The original list is : " + str(test_list))

# initializing K
K = 20

# enumerate to get indices of all inner and outer list
res = [ele[0] for sub in enumerate(test_list) for ele in enumerate(sub[1])]

# getting index of word
res = res[K]

# printing result
print("Index of character at Kth position word : " + str(res))

# Python3 code to demonstrate working of
# Word Index for K position in Strings List
# Using next() + zip() + count()

from itertools import count

# initializing list
test_list = ["geekforgeeks", "is", "best", "for", "geeks"]

# printing original list
print("The original list is : " + str(test_list))

# initializing K
K = 20

# count() for getting count
# pairing using zip()
cnt = count()
res = next(j for sub in test_list for j, idx in zip(
    range(len(sub)), cnt) if idx == K)

# printing result
print("Index of character at Kth position word : " + str(res))

# Python3 code to demonstrate working of
# Word Index for K position in Strings List
# Using nested loop

# initializing list
test_list = ["geekforgeeks", "is", "best", "for", "geeks"]

# printing original list
print("The original list is : " + str(test_list))

# initializing K
K = 20

# initializing index counter
idx = 0

# iterating over each word in the list
for word in test_list:
  
    # if the kth position is in the current word
    if idx + len(word) > K:
        
        # printing result
        print("Index of character at Kth position word : " + str(K - idx))
        break
    
    # if the kth position is not in the current word
    else:
        idx += len(word)

# if the kth position is beyond the end of the list
else:
    print("K is beyond the end of the list")