首页 文章

Python将2darray的最后一行和第二列添加到最后位置 - Polar Contour图

提问于
浏览 次
0

我创建了一个极坐标轮廓图并尝试通过将第一行的数据添加到结尾来关闭它 . 在这张照片中看到180度：PolarPlot

使用meshgrid和griddata模块创建数据 . 数组大小为nxn类型 . 例如：ri - float64 - （100,100）大小

print ri 
    [[ 0.00160738  0.00184056  0.00207375 ...,  0.23409252  0.23432571
   0.23455889]
 [ 0.00160738  0.00184056  0.00207375 ...,  0.23409252  0.23432571
   0.23455889]
 [ 0.00160738  0.00184056  0.00207375 ...,  0.23409252  0.23432571
   0.23455889]

theta和contour创建等效 .

绘图由matplotlib完成

我怎样才能做到这一点？这是在180度“关闭”情节的正确方法吗？

以下是情节片段：

fig4 = plt.figure()
ax = fig4.add_subplot(111)
ax = plt.axes(polar=True)
image=plt.contourf(thetai,ri/d,contourd,128,vmin=0,extent=([-math.pi,+math.pi,min(ri[0]/d),max(ri[0]/d)]),cmap=plt.cm.Paired)

ax.set_xlabel(r'$\Theta$')
ax.set_ylabel(r'$r$')
python arrays matplotlib

2 回答

  • 1

    从您的问题中，您的数据看起来并不清楚 . 一般来说，对于要关闭的极坐标图，数据的最后一点必须与第一点相同 . 因此，如果您有一个数组 X ，其中 X[:,0] 是角度而 X[:,1] 是半径，您可以通过将第一个元素附加到末尾来关闭极坐标图，如下所示：

    X_closed = np.append(X,X[[0]],axis = 0)

    即您只需将第一行添加到结尾，而不是第一列 .

    回复于
  • 2

    我不确定您的确切情节，但您可以使用numpy.pad执行您描述的数组操作 .

    我们可以给 np.pad 你的原始数组和 pad_width 的 ((0,1),(0,1)) ，这意味着左边有0个列，右边有1个，顶部有0个列，底部有1个 . 设置 mode='edge' 以复制数组边缘的值 .

    例如：

    In [16]: a = np.array([[1,1,5],[2,2,6],[7,8,9]])
In [17]: a
Out[17]: 
array([[1, 1, 5],
       [2, 2, 6],
       [7, 8, 9]])

In [18]: np.pad(a,((0,1),(0,1)),mode='edge')
Out[18]: 
array([[1, 1, 5, 5],
       [2, 2, 6, 6],
       [7, 8, 9, 9],
       [7, 8, 9, 9]])
    回复于

相关问题