Оптимизировать функцию заполнения столбца 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)

Новые блоги и статьи Все статьи Все блоги /
Установка Qt Creator для C и C++: ставим среду, CMake и MinGW без фреймворка Qt 8Observer8 05.04.2026 Среду разработки Qt Creator можно установить без фреймворка Qt. Есть отдельный репозиторий для этой среды: https:/ / github. com/ qt-creator/ qt-creator, где можно скачать установщик, на вкладке Releases:. . .	AkelPad-скрипты, структуры, и немного лирики.. testuser2 05.04.2026 Такая программа, как AkelPad существует уже давно, и также давно существуют скрипты под нее. Тем не менее, прога живет, периодически что-то не спеша дополняется, улучшается. Что меня в первую очередь. . .	Отображение реквизитов в документе по условию и контроль их заполнения Maks 04.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "ПланированиеСпецтехники", разработанного в конфигурации КА2. Данный документ берёт данные из другого нетипового документа. . .	Фото всей Земли с борта корабля Orion миссии Artemis II kumehtar 04.04.2026 Это первое подобное фото сделанное человеком за 50 лет. Снимок называют новым вариантом легендарной фотографии «The Blue Marble» 1972 года, сделанной с борта корабля «Аполлон-17». Новое фото. . .
Вывод диалогового окна перед закрытием, если документ не проведён Maks 04.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "СписаниеМатериалов", разработанного в конфигурации КА2. Задача: реализовать программный контроль на предмет проведения документа. . .	Программный контроль заполнения реквизитов табличной части документа Maks 02.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "СписаниеМатериалов", разработанного в конфигурации КА2. Задача: 1. Реализовать контроль заполнения реквизита. . .	wmic не является внутренней или внешней командой Maks 02.04.2026 Решение: DISM / Online / Add-Capability / CapabilityName:WMIC~~~~ Отсюда: https:/ / winitpro. ru/ index. php/ 2025/ 02/ 14/ komanda-wmic-ne-naydena/	Программная установка даты и запрет ее изменения Maks 02.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "СписаниеМатериалов", разработанного в конфигурации КА2. Задача: при создании документов установить период списания автоматически. . .

@Gdez 8851 / 4502 / 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

Решение