如何在matplotlib中标记一个点

时间:2013-11-30 22:50:12

标签: python matplotlib plot

我正在使用Matplotlib在3D空间中绘制表面图。为了使图更易读,我想标记曲面中的最低点(我已经使用数据找到了x,y坐标),显示其x,y,z值。另外,我想注释x,y,z轴,指定该轴上的值的名称。谁知道怎么做?非常感谢。

这是代码(我正在从文件中读取错误率数据):

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

def readdata(filename):
   reader = np.loadtxt(filename, dtype = np.ndarray, delimiter=',')
   return reader

fig = plt.figure()
ax = fig.gca(projection='3d')
Z = readdata('continuedData')
for i in range(len(Z)):
    Z[i] = float(Z[i])
Z = np.reshape(Z,(15,15))
X = np.arange(0.011, 0.026, 0.001)
Y = np.arange(0.11, 0.25, 0.01)
X, Y = np.meshgrid(Y, X)
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm,
      linewidth=0, antialiased=False)
ax.set_ylabel('learningRate')
ax.set_xlabel('momentum')
ax.set_zlabel('error rate')
ax.annotate('lowestPoint', xyz=(0.011,0.11,1.78199), xycoords='data',xytext=(10,10),
         textcoords='offset points', arrowprops=dict(arrowstyle="->"))
plt.show()

我正在尝试这样做但是ax.annotate有一个错误:

Traceback (most recent call last):
File "/home/yongfeng/workspace/ANN/plotContinuedData.py", line 25, in <module>
ax.annotate('lowestPoint', xyz=(0.011,0.11,1.78199), xycoords='data',xytext=         (10,10),textcoords='offset points', arrowprops=dict(arrowstyle="->"))
File "/usr/lib/pymodules/python2.7/matplotlib/axes.py", line 3348, in annotate
a = mtext.Annotation(*args, **kwargs)
TypeError: __init__() takes at least 3 arguments (6 given)

实际上我不知道是否需要指定Z值或者函数是否自动执行。所以我想注释情节中的左角。 enter image description here

1 个答案:

答案 0 :(得分:0)

尝试pylab.annotate()使用here

TypeError: __init__() takes at least 3 arguments (6 given)

这表示您使用的功能不正确。输入help(ax.annotate)