проверка существования индекса данных

Question

Когда я выполняю следующий код, он генерирует горизонтальную линию (по желанию), но только когда это местоположение (доступная позиция индекса) присутствует в 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:
      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')