Python:在同一图表中绘图

时间:2018-05-17 17:25:15

标签: python arrays list matplotlib

我有一个这样的数据集,但对于许多ID:

Information = [{'id' : 1,
'a' : array([0.7, 0.5 , 0.20 , 048 , 0.79]),
'b' : array([0.1, 0.5 , 0.96 , 08 , 0.7]))}, 
{'id' : 2,
'a' : array([0.37, 0.55 , 0.27 , 047 , 0.79]),
'b' : array([0.1, 0.5 , 0.9 , 087 , 0.7]))}]

我想在x轴上绘制一个图形,在y轴上绘制b,用于许多不同的ID。

我可以通过这样做一个情节?

a_info = information[1]['a'] 
b_info = information [2]['b]
plt.scatter(a_info , b_info) 
plt.show()

但我如何为所有情节做到这一点?

e = [d['id'] for d in information]
for i in e:
  a_info = information[i]['a'] 
  b_info = information [i]['b]
  plt.scatter(a_info , b_info) 
  plt.show()

2 个答案:

答案 0 :(得分:0)

您可以遍历id,并为每个子结构创建绘图:

import matplotlib.pyplot as plt
from numpy import array
information = [{'id' : 1, 'a':array([0.7, 0.5 , 0.20 , 0.48 , 0.79]), 'b':array([0.1, 0.5 , 0.96 , 0.8 , 0.7])}, {'id':2, 'a':array([0.37, 0.55, 0.27 , 0.47 , 0.79]), 'b':array([0.1, 0.5 , 0.9 , 0.87 , 0.7])}]
colors = iter(['b', 'g', 'r', 'c', 'm', 'y', 'k', 'w'])
for i in information:
   plt.scatter(i['a'], i['b'], label = 'id{}'.format(i['id']), color=next(colors))

plt.legend(loc='upper left')
plt.show()

enter image description here

答案 1 :(得分:0)

您可以遍历所有ID并绘制它们:

for i in Information:
    plt.scatter(i['a'], i['b'], label=i['id'])
plt.legend()
plt.show()

输出:

enter image description here