matplotlib x-axis wrong order

[SchroedingersLion]

Greetings!

I would like to create a simple 2D plot of 270 data points.

batchnrs = list(range(0,135))*2
plt.plot(batchnrs,  loss_per_batch, label="loss")
plt.show()

However, the x-axis in the plot does not go 0,1,...,134,0,1,...,134 (as I would like it to do) but only from 0 to 134 and the second half of y elements is simply plotted right across the first one. I would like to plot my data points as if the x-axis went from 0 to 269, so I just need to change what is portrayed on the x-ticks.


Best,
SL

[deanhystad]

Why not set the axis to 0..269? That will give you what you want. What you are seeing, with the second line starting at zero, is what I would expect to happen.

[nilamo]

Why not let the data drive the labels?

batchnrs = list(range(len(loss_per_batch)))
plt.plot(batchnrs,  loss_per_batch, label="loss")
plt.show()

[SchroedingersLion]

Thanks guys!

The thing is that the second half of my batches are batches from a following epoch. Earlier in the document, they were labelled from 0 to 134 again, which is why I would like my x-axis to be like 0,1,2,...,134,0,1,2,.... But there might be no straight forward way to do this..

[nilamo]

Maybe someone more familiar with matplotlib/pyplot will chime in, but I don't think you can have the same value in multiple places along an axis. Is it possible to give the second epoch unique labels?