在Python中绘制特定列表

Question

我正在用Python构建一个GUI，想要根据用户绘制目标来绘制员工的每日奖金：

Bob=[100,30,50,90]
Sammy=[10,60,90,200] 
Tom=[70,90,90,90]

# input from GUI User is Tom
ploting_target='Tom'

if ploting_target=='Tom':`
   plt.plot([0,1,2,3], Tom)

elif ploting_target=='Sammy':
   plt.plot([0,1,2,3], Sammy)
```
 plt.plot([0,1,2,3], Tom)
`
   

____________________________________________
#expecting  
#find_target=list_of_employee.index(ploting_target)

#plt(plot([0,1,2,3], list_of_employee[find_target])

Answer 1

你的问题不清楚。如果您有员工列表，您可以使用

Pandas

来组织数据（几乎与 @JohanC 使用

dict

建议的相同）：

# pip install pandas
import pandas as pd
import matplotlib.pyplot as plt

Bob=[100,30,50,90]
Sammy=[10,60,90,200] 
Tom=[70,90,90,90]

df = pd.DataFrame({'Bob': Bob, 'Sammy': Sammy, 'Tom': Tom})
df.index = 'Date ' + df.index.astype(str)

输出：

>>> df
       Bob  Sammy  Tom
Day 0  100     10   70
Day 1   30     60   90
Day 2   50     90   90
Day 3   90    200   90

现在你可以做

df.plot()

:

或针对特定员工

df['Sammy'].plot()

：

或者查看员工列表（但不是全部）

df[['Bob', 'Tom']].plot()

：