如何用python绘制x-y线图？

Question

你好，我有以下数据帧：

df = [{'Column1': 1, 'Colunm2': 'A', 'Colunm3': 2}, 
      {'Column1': 2, 'Colunm2': 'A', 'Colunm3': 4},
      {'Column1': 3, 'Colunm2': 'A', 'Colunm3': 1},
      {'Column1': 1, 'Colunm2': 'B', 'Colunm3': 7},
      {'Column1': 2, 'Colunm2': 'B', 'Colunm3': 2},
      {'Column1': 3, 'Colunm2': 'B', 'Colunm3': 9}]

如何在Column1和Colunm3之间绘制X Y图，但要绘制两条不同的内容线，一条用于值Colunm2 = 'A'，另一条用于值Colunm2 ='B'？你知道吗

Answer 1

你可以试试这个：

import matplotlib.pyplot as plt

df = [{'Column1': 1, 'Colunm2': 'A', 'Colunm3': 2},
      {'Column1': 2, 'Colunm2': 'A', 'Colunm3': 4},
      {'Column1': 3, 'Colunm2': 'A', 'Colunm3': 1},
      {'Column1': 1, 'Colunm2': 'B', 'Colunm3': 7},
      {'Column1': 2, 'Colunm2': 'B', 'Colunm3': 2},
      {'Column1': 3, 'Colunm2': 'B', 'Colunm3': 9}]

first = [(i["Column1"], i['Colunm3'])  for i in df if i['Colunm2'] == 'A']
second = [(i["Column1"], i['Colunm3']) for i in df if i['Colunm2'] == 'B']


plt.plot([a for a, b in first], [b for a, b in first])
plt.plot([a for a, b in second], [b for a, b in second])

plt.show()