I have a dataset that is of shape (n, x, 2). The length of x is fixed, such as 5. So for each n, I have an array of size (x, 2). For each n, I want to plot the first column