Matplotlib Line3DCollection 五彩线边缘“锯齿状”

Question

基于matplotlib 示例代码，我构建了一个多色线的 3D 版本。我正在使用 jupyter 笔记本，通过使用%matplotlib notebook我可以放大绘图，并且在我的浏览器中可以平滑地呈现角落边缘 - 完美！但是，当我将绘图导出为 png 或 pdf 文件以供进一步使用时，角边缘是“锯齿状的”。

任何想法如何平滑 3D 多彩线？

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap, BoundaryNorm    
from mpl_toolkits.mplot3d import Axes3D
from mpl_toolkits.mplot3d.art3d import Line3DCollection
%matplotlib notebook

# Generate random data
np.random.seed(1)
n = 20 # number of data points
#set x,y,z data
x = np.random.uniform(0, 1, n)
y  = np.random.uniform(0, 1, n)
z = np.arange(0,n)

# Create a colormap for red, green and blue and a norm to color
# f' < -0.5 red, f' > 0.5 blue, and the rest green
cmap = ListedColormap(['r', 'g', 'b'])
norm = BoundaryNorm([-1, -0.5, 0.5, 1], cmap.N)

#################
### 3D Figure ###
#################

# Create a set of line segments
points = np.array([x, y, z]).T.reshape(-1, 1, 3)
segments = np.concatenate([points[:-1], points[1:]], axis=1)

# Create the 3D-line collection object
lc = Line3DCollection(segments, cmap=plt.get_cmap('copper'),
                    norm=plt.Normalize(0, n))
lc.set_array(z) 
lc.set_linewidth(2)

#plot
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.set_zlim(0, max(z))
plt.title('3D-Figure')
ax.add_collection3d(lc, zs=z, zdir='z')

#save plot
plt.savefig('3D_Line.png', dpi=600, facecolor='w', edgecolor='w',
            orientation='portrait')

Answer 1

我认为连接样式是控制段接头外观的因素。Line3DCollection确实有一个set_joinstyle()功能，但这似乎没有任何区别。因此，我必须放弃Line3DCollection并逐段绘制线，并为每个线段调用其set_solid_capstyle('round').

以下是对我有用的：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

# Generate random data
np.random.seed(1)
n = 20 # number of data points
#set x,y,z data
x = np.random.uniform(0, 1, n)
y  = np.random.uniform(0, 1, n)
z = np.arange(0,n)


#################
### 3D Figure ###
#################

# Create a set of line segments
points = np.array([x, y, z]).T.reshape(-1, 1, 3)
segments = np.concatenate([points[:-1], points[1:]], axis=1)

cmap=plt.get_cmap('copper')
colors=[cmap(float(ii)/(n-1)) for ii in range(n-1)]

#plot
fig = plt.figure()
ax = fig.gca(projection='3d')

for ii in range(n-1):
    segii=segments[ii]
    lii,=ax.plot(segii[:,0],segii[:,1],segii[:,2],color=colors[ii],linewidth=2)
    #lii.set_dash_joinstyle('round')
    #lii.set_solid_joinstyle('round')
    lii.set_solid_capstyle('round')

ax.set_zlim(0, max(z))
plt.title('3D-Figure')

#save plot
plt.savefig('3D_Line.png', dpi=600, facecolor='w', edgecolor='w',
            orientation='portrait')

缩放时输出图像：