I\'ve got a dataframe like
xs = pd.DataFrame({ \'batch1\': { \'timestep1\': [1, 2, 3], \'timestep2\': [3, 2, 1] } }).T