При использовании df.drop в pandas смещение индекса строк

#python #pandas

#python #pandas

Вопрос:

Я работаю над программой, в которой я беру случайное значение, нахожу координаты, близкие к ним по расстоянию, добавляю их в группу и удаляю их из фрейма данных.

Итак, предполагая, что у меня есть список этих координат

введите описание изображения здесь

Они должны быть сгруппированы так, чтобы индекс

  • 0 и 1 вместе
  • 2 и 4 вместе
  • 3 один

Проблема, с которой я сталкиваюсь, заключается в том, что начальный индекс выбирается случайным образом для подсчета количества строк с использованием

 randInt = random.randint(0, count_row - 1)
 

и когда я удаляю координатную точку из фреймов данных, она не сдвигает остальные вверх, как показано на следующем рисунке.

введите описание изображения здесь

В настоящее время я удаляю точки, используя следующие команды.

Выбранный случайным образом индекс добавляется в список, а затем удаляется из фрейма данных с помощью

 df.drop(current, inplace=True)
 

Затем, если точка находится на определенном расстоянии, она добавляется в список и удаляется с помощью (где index — это индекс точки в фреймах данных при циклическом просмотре)

 df.drop(index, inplace=True)
 

Итак, мой вопрос в том, как я могу сделать так, чтобы при удалении точки она сдвигала идентификаторы вместе с ней?

Ответ №1:

Попробуйте сбросить индекс


df = df.reset_index(drop=True)

Комментарии:

1. Большое вам спасибо, это было что-то настолько простое, что я застрял на некоторое время!