数据框架索引存在检查

Question

当我执行以下代码时，它会生成水平行(根据需要(，但是只有在my_data中存在此位置(可用索引位置(时(MY_DATA过滤/以前选择了MY_DATA(。

plt.axhline(y=my_data.loc[6805], color='green', linestyle='dashed')

我想写类似的东西：

If my_data.loc[6805] is not None:
   plt.axhline(y=my_data.loc[6805], color='green', linestyle='dashed')

如何检查索引位置是否可用？

Answer 1

1]错误异常处理

• 您可以使用错误异常处理
实现此目的
• 如果该位置不存在，那么您将获得KeyError:

• try-except是您正在寻找的

  try:
      mplt.axhline(y=my_data.loc[6805], color='green', linestyle='dashed')
  except:
      pass
  

2]检查6805是否是索引 [仅当6805是索引时]

• 您可以做的其他事情是在6805中检查索引

  if 6805 in my_data.index.tolist():
      mplt.axhline(y=my_data.loc[6805], color='green', linestyle='dashed')
  

3]检查行的数量

• 您可以检查行总数是否大于或等于6805

  if my_data.shape[0] >= 6805 :
      mplt.axhline(y=my_data.loc[6805], color='green', linestyle='dashed')