0

我有一些 3D 数据，例如Zd=[x, y, z, f] 中的一列数字在哪里z，用作颜色信息。

f是一个标志，它是

  1. 0 如果x并且y有一些特定的值（丑^^）
  2. 1 如果x并且y没问题

所以对于好的数据d[ d[:,3] == 1 ] 我想生成一个配置文件

plt.imshow(resampled.T, extent=extent, vmin=MIN, vmax=MAX, origin='lower')

对于丑陋的数据d[ d[:,3] == 0 ]，我只想使用特定的颜色，例如黑色

有没有办法意识到这一点？

编辑：结合@eumiro 和@Rutger Kassies 的评论，我现在得到以下结果 在此处输入图像描述

我认为这是令人满意的。为了完整起见（或者可能有一些我不知道的优化^^），这里是代码和数据：

import numpy as np
from matplotlib.mlab import griddata
import matplotlib
import matplotlib.pyplot as plt


def plotprofile(x, y, z0, name='dummy', save=1):
    #plt.figure()
    N = 50j
    z = z0[:,0]
    extent = (min(x), max(x), min(y), max(y))
    xs,ys = np.mgrid[extent[0]:extent[1]:N, extent[2]:extent[3]:N]
    resampled = griddata(x, y, z, xs, ys)

    cmap = plt.get_cmap()
    cmap.set_bad(color = 'k', alpha = 1.)
    #plt.imshow(resampled.T, cmap='Greys', extent=extent, origin='lower', interpolation='spline36')
    plt.imshow(resampled.T,  cmap=cmap, extent=extent, origin='lower',  vmin=min(z),  vmax=-min(z),interpolation='spline36')

cbar=plt.colorbar()
s=20
plt.ylabel(r"$y$", size=s)
plt.xlabel(r"$x", size=s)
plt.xlim([x.min(),x.max()])
plt.ylim([y.min(),y.max()])

if save:
    for end in ["pdf", "png", "eps"]:
        print "save %s.%s"%(name,end)
        plt.savefig("%s.%s"%(name,end))
else:
    plt.show()
plt.clf()


if __name__ == '__main__':
    filename = 'data.txt'
    data = np.loadtxt(filename)
    x = data[:,0]
    y = data[:,1]
    z = data[:,3:]
    plotprofile(x, y, z,  'dummy', 0)
4

1 回答 1

0

你不能通过使用 f 来掩盖 z 来在 z 中创建一个正常的颜色图吗？

dd = d[:, :3]
dd[:,2] = dd[:,2] * d[:,3]

然后转换成这样的图像：

M = dd.max(0)
m = dd.min(0)

x = np.arange(m[0], M[0] + 1)
y = np.arange(m[1], M[1] + 1)
[X, Y] = np.meshgrid(x, y)

Z = np.zeros_like(X)

for num in range(0,size(dd, 0)):
    Z[dd[num, 0], dd[num, 1]] = dd[num, 2]

现在您应该能够Z像普通图像或表面一样绘制[X, Y]

于 2013-01-15T13:21:39.770 回答