I'm writing a script that will take as user inputed string, and print it vertically, like so:

input = "John walked to the store"output = J w t t so a o h th l   e on k     re     ed

I've written most of the code, which is as follows:

import sysdef verticalPrint(astring):wordList = astring.split(" ")wordAmount = len(wordList)maxLen = 0for i in range (wordAmount):length = len(wordList[i])if length >= maxLen:maxLen = length### makes all words the same length to avoid range errors ###for i in range (wordAmount):if len(wordList[i]) < maxLen:wordList[i] = wordList[i] + (" ")*(maxLen-len(wordList[i]))for i in range (wordAmount):for j in range (maxLen):print(wordList[i][j])def main():astring = input("Enter a string:" + '\n')verticalPrint(astring)main()

I'm having trouble figure out how to get the output correct. I know its a problem with the for loop. It's output is:

input = "John walked"output = Johnwalked

Any advice? (Also, I want to have the print command used only once.)

Use itertools.zip_longest:

>>> from itertools import zip_longest
>>> text = "John walked to the store"
for x in zip_longest(*text.split(), fillvalue=' '):print (' '.join(x))
...     
J w t t s
o a o h t
h l   e o
n k     re     ed