import numpy as np
import matplotlib.pyplot as plt

# Data
x = np.linspace(-10, 10, 10)
y = np.linspace(-10, 10, 10)
X, Y = np.meshgrid(x, y)
U = X
V = X**2

# basic streamline plot
plt.figure()
sp1 = plt.streamplot(x, y, U, V)

# Attempt to hide lines and arrows from streamline plot
plt.figure()
sp2 = plt.streamplot(x, y, U, V)
sp2.lines.set_visible(False)
sp2.arrows.set_visible(False)

plt.show()

import matplotlib.pyplot as plt
from matplotlib import patches, lines
from matplotlib import collections


def draw_test(ax, trans1, trans2, title):
    if trans1:
        transform1 = ax.transData
    else:
        transform1 = None
    if trans2:
        transform2 = ax.transData
    else:
        transform2 = None
    # Add arrow
    arrow_tail = [0, 0]
    arrow_head = [1, 0]
    p = patches.FancyArrowPatch(arrow_tail, arrow_head, arrowstyle='-|>',
                                mutation_scale=40, transform=transform1,
                                margins=False)
    ac = collections.PatchCollection([p], margins=False, transform=transform2)
    ax.add_collection(ac)
    # Add another kind of patch
    e = patches.Ellipse([0, 0], .2, .5, .5, transform=transform1)
    ec = collections.PatchCollection([e], margins=False, transform=transform2)
    ax.add_collection(ec)
    # Add lines
    streamline = [[0, -0.5], [0.5, 0], [1, 0.5]]
    lc = collections.LineCollection([streamline], margins=False,
                                     transform=transform2)
    ax.add_collection(lc)
    # 
    ax.set_title(title)
    ax.set_xlim(-1, 1)
    ax.set_ylim(-1, 1)


# Create subplot
fig, axs = plt.subplots(3, 2, figsize=(8, 8))
axs[0, 1].remove()

# Add as patches
ax1 = axs[0, 0]
transform1 = ax1.transData
arrow_tail = [0, 0]
arrow_head = [1, 0]
p1 = patches.FancyArrowPatch(arrow_tail, arrow_head, arrowstyle='-|>',
                             mutation_scale=40, transform=transform1,
                             margins=False)
ax1.add_patch(p1)
e1 = patches.Ellipse([0, 0], .2, .5, .5)
ax1.add_patch(e1)
streamline1 = [[0, -0.5], [0.5, 0], [1, 0.5]]
ax1.add_line(lines.Line2D(*zip(*streamline1)))
ax1.set_title("Added as patches")
ax1.set_xlim(-1, 1)
ax1.set_ylim(-1, 1)

# Add with various transform
draw_test(axs[1, 0], trans1=False, trans2=False, title="No transform used")
draw_test(axs[1, 1], trans1=True, trans2=False, title="Transform on patches")
draw_test(axs[2, 0], trans1=False, trans2=True, title="Transform on PatchCollection")
draw_test(axs[2, 1], trans1=True, trans2=True, title="Transform on both")

plt.tight_layout()
plt.show()