Фильтровать строки, содержащие список с непрерывными значениями в кадре данных pandas

Question

Привет! У меня есть датафрейм, показанный ниже:

    starttime                     endtime                                        positions
0   2019-05-16 05:34:26.870 2019-05-16 05:34:41.721 [7, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24...
1   2019-05-16 05:33:56.143 2019-05-16 05:34:10.995 [9, 11, 12, 15, 16, 17, 18, 19, 20, 21, 22, 23...
2   2019-05-16 05:33:35.659 2019-05-16 05:33:50.510 [13, 14, 15, 16, 17, 18, 19, 20, 21, 23, 24, 2...
3   2019-05-16 05:33:04.933 2019-05-16 05:33:19.784 [8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19,...
4   2019-05-16 05:34:11.507 2019-05-16 05:34:26.358 [3, 4, 9, 10, 11, 12, 15, 16, 17, 18, 19, 20, ...

Я хочу взять строки таким образом, чтобы список содержал непрерывные значения в форме list(range(min(val),max(val))).

Я пытался

df[df["positions"] == list(range(min(df["positions"],max(df["positions"]))))]

Но я получаю ошибку следующим образом:

ValueError: длины должны совпадать для сравнения

Это потому, что каждый список имеет разную длину? Как я могу решить это, если это делает?

Answer 1

Одним из способов является использование .apply в столбце списков:

df['position'].apply(lambda x: x == list(range(min(x), max(x) + 1)))

Минимальный пример

# Example input
df = pd.DataFrame({'starttime': list(range(3)), 
                   'endtime': list(range(1, 4)), 
                   'positions': None})

# Manually insert lists into the 'positions' column entries
df.iat[0, 2] = [1, 4, 9]
df.iat[1, 2] = list(range(6))
df.iat[2, 2] = list(range(-4, 3))

# Get a boolean Series
df['positions'].apply(lambda x: x == list(range(min(x), max(x) + 1)))

0    False
1     True
2     True