Actually axis could be a string ('index' or 'columns'), not only an integer. You could add string type here. Moreover you could check on typing stage of the implementation that axis type is able to be a string.

Originally posted by @densmirn in #227