Оптимизировать функцию заполнения столбца dataframe pandas

@Polarized_d · Регистрация: 12.11.2018

Студворк — интернет-сервис помощи студентам

Доброго времени суток. У меня есть функция, которая заполняет столбец second_event_time, для каждого пользователя отдельно, значением столбца date, при events_ordinal_number == 2. Функция рабочая и выполняет свою задачу. Но работает очень долго. Кажется, что заполнение столбца можно сделать в одну строку, но не представляю как. Обшарил русский и английский стак - тщетно.(может гуглил не правильно). Скрипт функции ниже.

Python
1
2
3
4
5
6
7
8
def calculate_second_event_time(df):
    
    for i in df['user_id'].unique():
        tmp = df.loc[(df['user_id'] == i) & (df['events_ordinal_number'] == 2)]['time']
        if tmp.count() == 1:
            df.loc[df['user_id'] == i, ['second_event_time']] = tmp.item()
​
    return df

@Gdez · 18.02.2022, 17:54

Polarized_d, Вместо кода (он ведь "медленный") лучше выложи пример - что есть и что должно получиться...

@Polarized_d · 18.02.2022, 18:00 **[ТС]**

Нету данных. Есть функция, которую надо оптимизировать, уменьшить время выполнения. Все что есть - это этот код.

@Gdez · 18.02.2022, 18:59

Polarized_d,

заполняет столбец second_event_time, для каждого пользователя отдельно, значением столбца date, при events_ordinal_number == 2.

Python
1
df['second_event_time'] = df.date[df.events_ordinal_number.values == 2]

Или

Python
1
df['second_event_time'] = np.where(df.events_ordinal_number.values == 2, df.date, np.nan)

@Polarized_d · 18.02.2022, 19:04 **[ТС]**

Это не то что делает моя функция. То что вы написали заполняет только одну строку для конкретного пользователя. А что если пользователь в таблице встречается несколько раз?

@Gdez · 18.02.2022, 23:07

Polarized_d,

Python
1
2
3
4
5
6
7
8
9
10
11
import pandas as pd
import numpy as np
 
df = pd.DataFrame({'user_id': [1,2,3,4,5,1,2,3,4,5], 'events_ordinal_number': [5,4,2,3,1,1,1,1,1,2], 'date': [11,22,33,44,55,66,77,88,99,100]})
print(df)
df1 = df[['user_id', 'date']][df.events_ordinal_number.values == 2]
df1.rename(columns = {'date' : 'second_event_time'}, inplace = True)
print(df1)
 
df = pd.merge(df, df1, on='user_id', how='left')
print(df)

Новые блоги и статьи Все статьи Все блоги /
Подключение Box2D v3 к SDL3 для Android: физика и отрисовка коллайдеров 8Observer8 29.01.2026 Содержание блога Box2D - это библиотека для 2D физики для анимаций и игр. С её помощью можно определять были ли коллизии между конкретными объектами. Версия v3 была полностью переписана на Си, в. . .	Инструменты COM: Сохранение данный из VARIANT в файл и загрузка из файла в VARIANT bedvit 28.01.2026 Сохранение базовых типов COM и массивов (одномерных или двухмерных) любой вложенности (деревья) в файл, с возможностью выбора алгоритмов сжатия и шифрования. Часть библиотеки BedvitCOM Использованы. . .	Загрузка PNG с альфа-каналом на SDL3 для Android: с помощью SDL_LoadPNG (без SDL3_image) 8Observer8 28.01.2026 Содержание блога SDL3 имеет собственные средства для загрузки и отображения PNG-файлов с альфа-каналом и базовой работы с ними. В этой инструкции используется функция SDL_LoadPNG(), которая. . .	Загрузка PNG с альфа-каналом на SDL3 для Android: с помощью SDL3_image 8Observer8 27.01.2026 Содержание блога SDL3_image - это библиотека для загрузки и работы с изображениями. Эта пошаговая инструкция покажет, как загрузить и вывести на экран смартфона картинку с альфа-каналом, то есть с. . .
Влияние грибов на сукцессию anaschu 26.01.2026 Бифуркационные изменения массы гриба происходят тогда, когда мы уменьшаем массу компоста в 10 раз, а скорость прироста биомассы уменьшаем в три раза. Скорость прироста биомассы может уменьшаться за. . .	Воспроизведение звукового файла с помощью SDL3_mixer при касании экрана Android 8Observer8 26.01.2026 Содержание блога SDL3_mixer - это библиотека я для воспроизведения аудио. В отличие от инструкции по добавлению текста код по проигрыванию звука уже содержится в шаблоне примера. Нужно только. . .	Установка Android SDK, NDK, JDK, CMake и т.д. 8Observer8 25.01.2026 Содержание блога Перейдите по ссылке: https:/ / developer. android. com/ studio и в самом низу страницы кликните по архиву "commandlinetools-win-xxxxxx_latest. zip" Извлеките архив и вы увидите. . .	Вывод текста со шрифтом TTF на Android с помощью библиотеки SDL3_ttf 8Observer8 25.01.2026 Содержание блога Если у вас не установлены Android SDK, NDK, JDK, и т. д. то сделайте это по следующей инструкции: Установка Android SDK, NDK, JDK, CMake и т. д. Сборка примера Скачайте. . .

@Gdez 8849 / 4501 / 1864 Регистрация: 27.03.2020 Сообщений: 7,317
	18.02.2022, 17:54
	Polarized_d, Вместо кода (он ведь "медленный") лучше выложи пример - что есть и что должно получиться... 0

@Polarized_d 0 / 0 / 0 Регистрация: 12.11.2018 Сообщений: 3
	18.02.2022, 18:00 [ТС]
	Нету данных. Есть функция, которую надо оптимизировать, уменьшить время выполнения. Все что есть - это этот код. 0

@Polarized_d 0 / 0 / 0 Регистрация: 12.11.2018 Сообщений: 3
	18.02.2022, 19:04 [ТС]
	Это не то что делает моя функция. То что вы написали заполняет только одну строку для конкретного пользователя. А что если пользователь в таблице встречается несколько раз? 0

Оптимизировать функцию заполнения столбца dataframe pandas

Решение