使用来自df

Question

我正在从已经加载了csv文件的df创建散点图。我正在尝试编写一个名为 scatter_plot（df，stock_name，x_data，y_data）的函数，该函数将为特定的股票名称提取x和y列。因此，数据框具有一个名为stock的列，该列具有许多不同的库存。

到目前为止，我有

def scatter_plot(df,stock_name,x_data,y_data):
  fig, ax = plt.subplots()
  x = df[x_data]
  y = df[y_data]
  ax.scatter(x, y)   
  ax.set_xlabel(x_data)    
  ax.set_ylabel(y_data)
  return ax

，我可以得到一个散点图，但是它为ENTIRE列x和y而不是特定股票绘制了散点图。如何将参数 stock_name 合并到函数中，以便只绘制特定股票的x和y？

Answer 1

当您的数据看起来像来自熊猫时，您将进行基于熊猫的归约

df_for_ibm = df[df.stock_name='IBM']

然后图表...在图表上只会显示“ IBM”。