0

我正在研究将 Python 与 MySQLdb 和 Matplotlib 一起使用。我正在寻找基于 ginput 图在 matplotlib 散点图中使用查询的值。我有以下工作：

import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np
from pylab import *
import random
import MySQLdb as mdb
import sys
from collections import defaultdict


##### Start the query ####
db = mdb.connect('localhost', 'root', 'password', 'xbee')
start = raw_input("Enter Start Date: ")

part_1 = "SELECT XBEE_ADDRESS_AL, XBEE_TEMPERATURE FROM xbeereadings WHERE Date='"
part_2 = start
part_3 = "'"
query_1 = part_1 + part_2 + part_3
cur = db.cursor()
cur.execute(query_1)

s = cur.fetchall()
print s
d = defaultdict(list)
for k, v in s:
    d[k].append(v)

i = 0
temp = [item[i] for item in d.values()]

figure(figsize=(15, 8))
img = mpimg.imread('floor.png')
imgplot = plt.imshow(img, cmap=cm.hot)
print "Left click to plot the sensors point on the image - Middle Click to remove the last point - Right click to End plotting"

# pts would be used with ginput to collect the place the sensor would be located. It returns the example array below
pts = ginput(n=0, timeout=0, mouse_add=1, mouse_pop=2, mouse_stop=3)

x = map(lambda x: x[0],pts) # Extract the values from pts
y = map(lambda x: x[1],pts) 
t = temp

result = zip(x,y,t)

img = mpimg.imread('floor.png')
imgplot = plt.imshow(img, cmap=cm.hot, vmin=-20, vmax=40)
scatter(x, y, marker='h', c=t, s=150, vmin=-20, vmax=40) #add colour c=?
print t

# Add cmap
colorbar()
show()

编辑：我得到了问题的前一部分（如何使用查询值作为 cmap 值），如新代码所示。我已经测量了温度（除以 100 以获得有效数字），然后将其放在图中。

我现在想要一些帮助/代码/起点的问题是：

1 - 如何从查询中将 ginput 点分配给传感器 ID？图上将放置 3 个传感器，因此我想将 id 和温度分配给一个点。我遇到的问题是它将 t 的第一个值分配给第一个点 - 将 t 的第二个值分配给第二个点。如何设置分配给特定点的温度值？

如果我在 1 小时内全部获取，它将为同一个传感器提供多个值。我想要某种时间控制，我可以为每个传感器绘制第一组结果 - 然后按下一个按钮，然后绘制每个传感器的下一个结果。它们都将同时运行，因此每个传感器 id 总会有一个值要绘制。

它也给出了这个错误 -

C:\Python27\lib\site-packages\matplotlib\colorbar.py:808: RuntimeWarning: invali
d value encountered in divide
  z = np.take(y, i0) + (xn-np.take(b,i0))*dy/db
Traceback (most recent call last):
  File "heatmap2.py", line 51, in <module>
    show()
  File "C:\Python27\lib\site-packages\matplotlib\pyplot.py", line 143, in show
    _show(*args, **kw)
  File "C:\Python27\lib\site-packages\matplotlib\backend_bases.py", line 108, in
 __call__
    self.mainloop()
  File "C:\Python27\lib\site-packages\matplotlib\backends\backend_tkagg.py", lin
e 69, in mainloop
    Tk.mainloop()
  File "C:\Python27\lib\lib-tk\Tkinter.py", line 325, in mainloop
    _default_root.tk.mainloop(n)
KeyboardInterrupt

那是因为两个值相同，所以 cmap 只有 1 个值？如果我将查询中的一个温度设置为 0.56，它就会消失。

我希望这是有道理的

4

1 回答 1

2

您遇到了ScalarMappables. 他们负责将数据标准化为范围 [0, 1] 并将该值传递给颜色映射。默认情况下，它将范围的底部设置为min(values_you_are_mapping)最大值，将顶部设置为最大值，如果您的所有值都相同，则会导致范围的宽度为零，并且映射会(v - max_v) / (max_v - min_v)爆炸。解决方案是告诉它范围应该是多少

imshow(..., vmin=min_t, vmax=max_t)
scatter(..., vmin=min_t, vmax=max_t)

在哪里max_t，min_t是你能得到的最高和最低温度。这也将使所有图形的颜色映射保持一致。

于 2013-04-09T15:07:15.610 回答