How do I plot a series of points in lines in matplotlib?

I am using matplotlib and Python 2.7

I have an MxN matrix of tuples, x-coordinate and velocity. How do I plot M lines of points with N points in each line with the specified x-coordinates? Preferably with the first row up?

I've tried various examples from the documentation, but to be honest I haven't found anything.

Here is an example I want to follow, t-coordinate goes from 0 to M, x range is fixed size. The points are placed on a horizontal line according to their values. Is it a little readable? Plot mockup


source to share

2 answers

It looks like you have something like this:

import numpy as np
import matplotlib.pyplot as plt

x = np.random.random((10, 20))
x = x.cumsum(axis=1)

fig, ax = plt.subplots()
for i, row in enumerate(x):
    ax.plot(row, i * np.ones_like(row), 'ko')

ax.set_ylim([-0.5, 9.5])


enter image description here


@EMS is absolutely right, I missed a rather key point of your question.

However, if you have nested lists of tuples, just convert them to an array. This will be a 3D array that you can slice as you need for x-position and speed. There is absolutely no need to generate a second dataset, and matplotlib will convert whatever you enter into it into a numpy array, no matter if there is a performance penalty.


import numpy as np

data = [[(1, 2), (3, 4)],
        [(5, 6), (7, 8)]]

data = np.array(data)

x = data[:,:,0]
velocity = data[:,:,1]


This gives:

array([[1, 3],
       [5, 7]])

array([[2, 4],
       [6, 8]])




Here is a Python script that makes fake data that should look like yours.

import numpy as np
import matplotlib.pyplot as plt

num_rows = 7
num_cols = 10

# Make a fake data array that just a list of lists.
# And each list has num_cols number of different tuples.
# The x-data is assumed to be the first coordinate of the
# tuple

my_data = []
for ii in range(num_rows):
    for jj in range(num_cols):
        my_data[ii].append( (24*np.random.rand(),np.random.rand()) )

# Now plot the different rows as separate plots.
fig = plt.figure()
ax = fig.add_subplot(111)
for ii in range(num_rows):

    # The y-axis values are just a constant based on the current row.
    cur_tvals = [ii]*num_cols

    # The x values are gotten by using a list comprehension to
    # grab just the first tuple element. 
    cur_xvals = [tup[0] for tup in my_data[ii]]

    # Add the current curve to the plot. Specifying '.' as the
    # symbol get rid of any lines connecting the markers.

# Setting axes based on num_rows
ax.set_ylim([-0.5, num_rows-1+0.5])


The points you want are displayed here: plot as requested



All Articles