Merged streamline examples by patniharshit · Pull Request #8336 · matplotlib/matplotlib · GitHub

Based on discussion in #8155 a new pull request combining work with #8082.
Refs #7956

This looks mostly good, but could you merge the figures into one using several subplots?

Can you remove this line? It is not informative.

I don't know why the one wth masking is behaving like this. Can you point out how to fix it?

It is because of imshow() applying an equal aspect ratio.

Looks like

I prefer the appearance of the plot with an equal aspect ratio. To get this in a desirable way, I'd used the gridspec module (it's always nice to show that off in other places)

fig = plt.figure(figsize=(7, 9))
gs = gspec.GridSpec(nrows=3, ncols=2, height_ratios=[1, 1, 2])
ax0 = fig.add_subplot(gs[0, 0])
ax1 = fig.add_subplot(gs[0, 1])
ax2 = fig.add_subplot(gs[1, 0])
ax3 = fig.add_subplot(gs[1, 1])
ax4 = fig.add_subplot(gs[2:, :])
ax4.set_aspect('equal')

They're blue, not red.

Done.

If I'm not mistaken, these are the same values as the ones at the top. Maybe give the ones for ax3 a different name, then you won't need to recreate these same values again here.

Oops, this comment is supposed to be about lines 62-65.

No, U,V are being changed at line 47-49.

In overall the PR seems nice to me, thanks for working on this @patniharshit :)! I still have some comments and suggestion. Some are taken from #8082. Again, if you are a native English speaker and disagree with some of my remarks about phrasing: drop them ;).

FWIW, switching from cmap=plt.cm.name_of_a_cmap to cmap='name_of_a_cmap' is just a suggestion, but I think it may make the script a (very little) bit simpler.

If I am correct, you will also need to change the references in the docs that points to the deleted files. Looking at #8082, it should now be:

in doc/users/screenshots.rst , l. 74: .. plot:: mpl_examples/images_contours_and_fields/streamplot_demo.py
in doc/users/prev_whats_new/whats_new_1.2.rst , l. 155: .. plot:: mpl_examples/images_contours_and_fields/streamplot_demo.py

stream plot <- streamplot?

Done.

stream line <- streamline?

Done.

start points <- starting points?

Done.

plt.cm.gray <- 'gray'?

Done.

plt.cm.autumn <- 'autumn'?

Done.

plt.cm.autmn <- 'autumn'?

Done.

Any reason for changing the U, V compared to the former streamplot_demo_start_points.py script? How bad would it be to simply re-use the same X, Y, U and V as for the other subplots? This way, you'd just need to create these (dummy) data at the beginning of the script (NB: it is already the case at ll. 18-22) and after that, the script is focused on the plotting aspect of streamplot. I think that it is also close to what @QuLogic had in mind in one of his remarks.

I have kept U, V exactly same as they are in previous streamplot_demo_start_points.py.

But if I comment out above lines and use same U,V everywhere then I get this plot -

Oh I think that I understand why the current situation was weird for both of us!

The current example in the official docs is indeed the one that you are refering to. However, the starting point example was changed in #6672 (by myself, which is why I was having the feeling that you changed U and V here) to use the same data as the other streamplot examples, which I think is really more demonstrative to show the different plotting options of streamplot. The related version of the example is only in our devdocs gallery (I guess because the PR was milestoned for 2.1 and not backported into 2.0). So I would definitively support to use the same data (built once for all at the top of the script) for all the subplots :).

If one decides to always plot the same data among all the different subplots, the 5 lines above could be deleted.

Huh, why is this comment here?! This is the opposite of what I had in mind ^^... Actually lines 19-23 would be the only place where I would suggest to create all the dummy data!

I am not sure that it is a mandatory requirement, but you may want to add a blank line betwenn the second ===== line and the first line of the paragraph. Maybe @NelleV would be able to confirm what SG and rST really require :).

I don't think it is.

Then one remaining argument in favor of adding a blank line would be consistency among our example scripts, as a lot of other examples in the docs seem to have one such blank line ;). But I am also fine with no blank line if SG and rST are OK!

Blank line added to maintain consistency among all examples.

need to re-review

I forgot to comment on this one during my review, sorry. Is this line really needed (I do not see the equivalent for the previous subplots)? If indeed it is and one actually uses the same dummy data accross all the subplots, I would then rather suggest using ax3.axis((-w, w, -w, w)) instead (if I understand correctly the purpose of this line).

Other subplots were going from -3 to 3 but this one was not, so probably this line was put there to have uniformity in all plots. I have kept this. But it can be deleted considering this is supposed to be an example, thus it should use minimal code to achieve target.

You are right. Due to the reduced amount of streamlines that are plotted, it seems like the axis is not exactly the same as for the other plots. So it might be wiser to keep it :). 👍 for the new version ax3.axis((-w, w, -w, w)).

A few remaining comments, among which my biggest one will be to use the same dummy data (defined at the beginning of the script) for all the subplots. But apart from that, the PR starts to look really good to me :).

Huh, why is this comment here?! This is the opposite of what I had in mind ^^... Actually lines 19-23 would be the only place where I would suggest to create all the dummy data!

Varying Color?

Done.

Plural form (as there are several streamlines), or "Streamplot with Masking"?

Changed to "Streamplot with masking".

Again, when it comes to English, I may be totally wrong, but I think "the streamplot function" may be more correct than the current "the stream plot function", as it is refering to the streamplot function and not to a stream plot. I guess I was not clear enough: it was actually the meaning of my previous comment ^^. However, for the other utterances of "stream plot"/"stream line," I will let a native English speaker (who may be you!) decide if it is better than "streamplot"/"streamline" or not.

I think generally, "streamline" is a word and "streamplot" is not, unless you're referring to the function itself (as in this case.)

@NelleV Should this be renamed as plot_something.py?

I don't think example deletion should be backported, so I've upped the milestone.

Travis failures seem unrelated (they are about test_mixedsubplots…)

Last tiny nitpicking, based on @QuLogic comment about writing “streamline” with a single word (basically it reverts some of the changes made by a530e79).

Besides, pinging @NelleV about the other @QuLogic's comment: should this be renamed as plot_something.py (for Sphinx gallery sake IIUC)?

Apart from that, the new version of the example script looks fine, thanks @patniharshit :)!

Question to core devs who may pass by: if @patniharshit is comfortable with git, would it be relevant to rebase, and squash a bit the commits?

streamline

All done.

streamline

streamlines

streamline

streamlines

Streamlines

LGTM :).

A few questions to other devs:

should this be renamed as “plot_something.py” for Sphinx gallery sake? (@NelleV , pinging our SG expert if she is around)
should it be rebased and some of the commits squashed if @patniharshit is comfortable with git?

@afvincent my answer is "yes" to both of those.

Renamed, rebased and commits squashed.

Minor fixes needed. Apart from that, the other Travis failures seems to be spurious ones.

'streamplot_demo.py' should be changed into 'plot_streamplot.py', shouldn't it?

Done.

Here too, 'streamplot_demo.py' should be changed into 'plot_streamplot.py', shouldn't it?

Done.

Oh forgot this. Will do it asap.

LGTM. Thanks @patniharshit :) !

Ugh, I was going to merge this, but it looks like it needs a rebase with that sphinx-gallery PR merged.

I'm going to attempt a rebase and merge

files

Okay, that looks like it worked, but since it's the first time I've done a rebase and force-push I'll wait for CI to pass before merging.

I see that the current version of the plot_streamplot.py from repo is giving errors/not executing.

I see no errors in the latest CI build; note that examples from master are not guaranteed to work on any released version.

How can I also replicate it? I am spending time on fixing the error from master. Could you please let me know, how to perform such a check?

Replicate what? There is no error in master.

I get the following error:

Traceback (most recent call last):
  File "plot_streamplot.py", line 50, in <module>
    cmap='autumn', start_points=seed_points.T)
  File "/usr/lib/python2.7/dist-packages/matplotlib/__init__.py", line 1814, in inner
    return func(ax, *args, **kwargs)
  File "/usr/lib/python2.7/dist-packages/matplotlib/axes/_axes.py", line 4464, in streamplot
    zorder=zorder)
  File "/usr/lib/python2.7/dist-packages/matplotlib/streamplot.py", line 140, in streamplot
    sp2[:, 0] += np.abs(x[0])
ValueError: operands could not be broadcast together with shapes (6,) (100,) (6,)

The syntax for streamline says that x and y should be 1D array and u and v as 2D array. However in the example, both x and y are made as 2D grids using np.mgrid. When they are made just as 1D array, the issue is fixed.

@pssk1988 -- are you working with a matplotlib installed from the latest github master?

@phobson : I don't think so :/ I will look for using the github matplotlib.

I am really sorry to bother you all without using the github version of matplotlib. Now it works fine.

afvincent added the Documentation label Mar 19, 2017

NelleV previously requested changes Mar 19, 2017

View reviewed changes

dstansby added this to the 2.0.1 (next bug fix release) milestone Mar 24, 2017

QuLogic reviewed Mar 24, 2017

View reviewed changes

QuLogic mentioned this pull request Mar 25, 2017

merge streamplot_demo_features streamplot_demo_masking streamplot_demo_start_points #8155

Closed

afvincent requested changes Mar 25, 2017

View reviewed changes

afvincent mentioned this pull request Mar 25, 2017

Merged and improved the streamplot demonstration #8082

Closed

afvincent reviewed Mar 25, 2017

View reviewed changes

afvincent requested changes Mar 25, 2017

View reviewed changes

QuLogic modified the milestones: 2.1 (next point release), 2.0.1 (next bug fix release) Mar 25, 2017

anntzer mentioned this pull request Mar 25, 2017

Suggested improvements to the gallery #7956

Closed

16 tasks done

afvincent requested changes Apr 2, 2017

View reviewed changes

afvincent approved these changes Apr 5, 2017

View reviewed changes

afvincent changed the title Merged streamline examples [MRG+1] Merged streamline examples Apr 5, 2017

patniharshit force-pushed the mergedStreamline branch from 3302135 to 7baa01e Compare April 5, 2017 20:24

afvincent requested changes Apr 10, 2017

View reviewed changes

afvincent approved these changes Apr 11, 2017

View reviewed changes

patniharshit added 4 commits April 12, 2017 11:49

Merge all figure in streamplot examples in one plot

bf2b43b

Add gridspec to have equal aspect ratio and remove references of deleted

92b1196

files

Remove data regeneration

913fdb7

Update example name in docs

d56b5ce

dstansby force-pushed the mergedStreamline branch from fe5dc72 to d56b5ce Compare April 12, 2017 11:08

dstansby self-assigned this Apr 12, 2017

dstansby merged commit b81576c into matplotlib:master Apr 12, 2017

dstansby changed the title [MRG+1] Merged streamline examples Merged streamline examples Apr 12, 2017

patniharshit deleted the mergedStreamline branch April 12, 2017 11:59

dstansby removed their assignment May 27, 2017

+              Y, X = np.mgrid[-w:w:100j, -w:w:100j]
+              U = -1 - X**2 + Y
+              V = 1 + X - Y**2
+              speed = np.sqrt(U*U + V*V)

+              Streamplot
+              ==========
+              A streamplot, or streamline plot, is used to display 2D vector fields. This
+              example shows a few features of the stream plot function:

+                  * Varying the color along a streamline.
+                  * Varying the density of streamlines.
+                  * Varying the line width along a stream line.
+                  * Controlling the start points of streamlines.

+              # Controlling the starting points of the streamlines
+              X, Y = (np.linspace(-3, 3, 100),
+                      np.linspace(-3, 3, 100))
+              U, V = np.mgrid[-3:3:100j, 0:0:100j]

+              Y, X = np.mgrid[-w:w:100j, -w:w:100j]
+              U = -1 - X**2 + Y
+              V = 1 + X - Y**2
+              speed = np.sqrt(U*U + V*V)

+              fig.colorbar(strm.lines)
+              ax3.set_title('Controlling Starting Points')
+              # Displaying the starting points with red symbols.

+              ax4.set_title('Streamline with Masking')
+              ax4.imshow(~mask, extent=(-w, w, -w, w), alpha=0.5,
+                        interpolation='nearest', cmap=plt.cm.gray, aspect='auto')

+              ax3 = fig.add_subplot(gs[1, 1])
+              strm = ax3.streamplot(X, Y, U, V, color=U, linewidth=2,
+                                   cmap=plt.cm.autumn, start_points=seed_points.T)

+              # Varying color along a streamline
+              ax1 = fig.add_subplot(gs[0, 1])
+              strm = ax1.streamplot(X, Y, U, V, color=U, linewidth=2, cmap=plt.cm.autumn)

+              # Displaying the starting points with blue symbols.
+              ax3.plot(seed_points[0], seed_points[1], 'bo')
+              ax3.axis((-3, 3, -3, 3))

+              A stream plot, or stream line plot, is used to display 2D vector fields. This
+              example shows a few features of the streamplot function:
+                  * Varying the color along a stream line.

+                  * Varying the density of stream lines.
+                  * Varying the line width along a stream line.
+                  * Controlling the starting points of stream lines.
+                  * Stream lines skipping masked regions and NaN values.

               local intensity of the vector field.
-              .. plot:: mpl_examples/images_contours_and_fields/streamplot_demo_features.py
+              .. plot:: mpl_examples/images_contours_and_fields/streamplot_demo.py

Merged streamline examples by patniharshit · Pull Request #8336 · matplotlib/matplotlib · GitHub

Navigation Menu

Repository navigation

Merged streamline examples - #8336

Merged streamline examples#8336dstansby merged 4 commits intomatplotlib:mastermatplotlib/matplotlib:masterfrom patniharshit:mergedStreamlinepatniharshit/matplotlib:mergedStreamlineCopy head branch name to clipboard

Conversation

patniharshit commented Mar 19, 2017

Uh oh!

NelleV left a comment

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

patniharshit commented Mar 20, 2017

Uh oh!

WeatherGod commented Mar 20, 2017 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

patniharshit commented Mar 21, 2017 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

phobson commented Mar 21, 2017

Uh oh!

patniharshit commented Mar 23, 2017

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

afvincent left a comment

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

patniharshit Mar 25, 2017 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Merged streamline examples#8336
dstansby merged 4 commits into
matplotlib:masterfrom
patniharshit:mergedStreamline

WeatherGod commented Mar 20, 2017 •
edited
Loading

patniharshit commented Mar 21, 2017 •
edited
Loading

patniharshit Mar 25, 2017 •
edited
Loading