Как улучшить мой огромный код для нового столбца с повторяемым максимумом с помощью Pandas?

Question

Я хочу получить столбец, который будет содержать обновляемый максимум с каждым новым значением, которое я написал в коде, но это очень неудобно, может быть, в Pandas есть способ проще?

data =pd.DataFrame({
'country': ['USA', 'USA', 'Belarus', 'Russia','Brazil'],
'population': [1, 25, 2, 100, 100000]})
for i in data['population']:
    # check index array now
    index_now = data.index[data.population == i]
    print(data.loc[:index_now[0]]['population'].max())

Я ожидаю+ новый столбец «итеративный максимум» с [1, 25, 25, 100, 100000] Этот максимум этого + все предыдущие столбцы.

Answer 1

Функция, которую вы ищете: кумулятивный максимум , и она уже встроена.

data['iterable maximum'] = data.population.cummax()