使用索引进行散点图图例

Question

我创建了一个包含一些股票信息的DataFrame。当我尝试使用索引作为数据标签时，它不起作用。这使我无法区分股票，特别是当我添加更多股票时。当我绘制图例时，它会显示索引列表，dtype和名称。它似乎将每个点组合成一个标签。

我的表：

         Dividend  ExpenseRatio  Net_Assets  PriceEarnings  PriceSales
Ticker
ijr       0.0142          0.12        18.0          20.17        1.05
ijh       0.0159          0.12        27.5          20.99        1.20

我的密码：

plt.scatter( df.PriceSales, df.PriceEarnings, label = df.index)
plt.xlabel('PriceSales')
plt.ylabel('PriceEarnings')  
plt.legend() 
plt.show() 

我的传奇输出：

Index(['ijr', 'ijh'],dtype='object',name='Ticker')

Answer 1

我可能错了，但我认为当您的索引是您想要单独绘制的唯一文本标签时，label=df.index将无效。如果您的索引是唯一的（或者它是一个groupby），您可以使用for循环在单个图中绘制单个股票行，并为它们提供唯一的颜色（c=np.random.rand(3,1)）

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

df = pd.DataFrame({'Dividend': [0.0142, 0.0159], 'ExpenseRatio': [0.12, 0.12],
                   'Net_Assets': [18.0, 27.5], 'PriceEarnings': [20.17, 20.99],
                   'PriceSales': [1.05, 1.2]},
                  columns=['Dividend', 'ExpenseRatio', 'Net_Assets', 'PriceEarnings', 'PriceSales'],
                  index=['ijr', 'ijh'])

df.index.name = 'Ticker'

for ticker,row in df.iterrows():
  plt.scatter(row['PriceSales'], row['PriceEarnings'], label=ticker, c=np.random.rand(3,1), s=100)

plt.xlabel('PriceSales')
plt.ylabel('PriceEarnings')
plt.legend()
plt.show()

结果：