6 ZigZag Conversion

The string "PAYPALISHIRING" is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)

P A H N

A P L S I I G

Y I R

And then read line by line: “PAHNAPLSIIGYIR”

Write the code that will take a string and make this conversion given a number of rows:

string convert(string text, int nRows);

convert(“PAYPALISHIRING”, 3) should return “PAHNAPLSIIGYIR”.

class Solution(object):
def convert(self, s, numRows):
"""
:type s: str
:type numRows: int
:rtype: str
"""
ans=""
for i in range(numRows):
j=i
jump = 0
while j<len(s):
ans = ans + s[j]
if numRows == 1:
j=j+1
elif (i!=numRows-1 and jump%2==0) or (i == 0) :
j=j+(numRows-i)*2-2
else:
j=j+(i+1)*2-2
jump=jump+1
return ans