python – 在轴之间移动集合

2019年7月28日 175次阅读

在玩
ImportanceOfBeingErnest’s code以在轴之间移动艺术家时,我认为将其扩展到集合(例如由plt.scatter生成的PathCollections)也很容易.没有这样的运气：

import matplotlib.pyplot as plt
import numpy as np
import pickle

x = np.linspace(-3, 3, 100)
y = np.exp(-x**2/2)/np.sqrt(2*np.pi)
a = np.random.normal(size=10000)

fig, ax = plt.subplots()
ax.scatter(x, y)

pickle.dump(fig, open("/tmp/figA.pickle", "wb"))
# plt.show()

fig, ax = plt.subplots()
ax.hist(a, bins=20, density=True, ec="k")

pickle.dump(fig, open("/tmp/figB.pickle", "wb"))
# plt.show()

plt.close("all")

# Now unpickle the figures and create a new figure
#    then add artists to this new figure

figA = pickle.load(open("/tmp/figA.pickle", "rb"))
figB = pickle.load(open("/tmp/figB.pickle", "rb"))

fig, ax = plt.subplots()

for figO in [figA, figB]:
    lists = [figO.axes[0].lines, figO.axes[0].patches, figO.axes[0].collections]
    addfunc = [ax.add_line, ax.add_patch, ax.add_collection]
    for lis, func in zip(lists, addfunc):
        for artist in lis[:]:
            artist.remove()
            artist.axes = ax
            # artist.set_transform(ax.transData) 
            artist.figure = fig
            func(artist)

ax.relim()
ax.autoscale_view()

plt.close(figA)
plt.close(figB)
plt.show()

产量

《python – 在轴之间移动集合》

删除artist.set_transform(ax.transData)(至少在调用ax.add_collection时)似乎有点帮助,但请注意y偏移仍然是关闭的：
《python – 在轴之间移动集合》
如何将集合从一个轴正确移动到另一个轴？

最佳答案 分散符具有IdentityTransform作为主变换.数据变换是内部偏移变换.因此,人们需要分别处理散射.

from matplotlib.collections import PathCollection
from matplotlib.transforms import IdentityTransform

# ...

if type(artist) == PathCollection:
    artist.set_transform(IdentityTransform())
    artist._transOffset = ax.transData
else:
    artist.set_transform(ax.transData)

不幸的是,没有set_offset_transform方法,因此需要替换._transOffset属性来设置偏移变换.

《python – 在轴之间移动集合》

点赞