Как итерироваться по датафрейму?

Question

[Timebird]

Имеется датафрейм data (на картинке).
Хочу: заменить type {} в значениях каждого из столбика на просто {}. Например, type 5 на 5.
Как планировал реализовать: сделать генератор, который пробегает по значениям каждого столбика и по split'у оставляет только число.

Реализация:

columns_names = data.columns

def columns_gen(data, column_name):
    for value in data[column_name].values:
        if type(value) == str:
            value = value.split()[1]
            yield value

for column_name in columns_names:
    columns_gen(data, column_name)

Но ничего не меняется, датафрейм остается таким же, как на картинке. Почему?
Заранее спасибо!

Comments

[Игорь]

Так ведь значения, которые возвращает генератор, вы нигде не сохраняете.

[Timebird]

Игорь, Да! Честно говоря, даже не заметил. Но как сохранить?
Пробую просто заменять значения ячейки на каждой итерации цикла:
value.replace(value.split()[1])
Но что-то да, эффекта тоже нет

[Animkim]

Timebird, Потому что вы снова не присваиваете новое значение.

Answer 1

[Игорь]

Попробуйте:

for column_name in data.colums:
    for i in len ( data.values ) :
        if type ( data [ column_name ] .values [ i ] == str :
            data [ column_name ] .values [ i ] = data [ column_name ] .values [ i ] .split() [ 1 ]

Comments

[Timebird]

Игорь, попробовал. Что-то у вас не так... У меня Ваш листинг завис на исполнении. :(
Не вижу, где здесь генераторы (я сам в них профан, вот сейчас сам буквально первый раз пытаюсь их применить, оттого и вопрос)

[Игорь]

Timebird, я дурак:

for column_name in data.columns:
    for i in len ( data [ column_name ] .values ) :
        if type ( data [ column_name ] .values [ i ] == str :
            data [ column_name ] .values [ i ] = data [ column_name ] .values [ i ] .split() [ 1 ]

Пишу без генераторов, помоему они здесь не нужны.

[Timebird]

Игорь, они нужны, если ничего не путаю, если значений в массиве очень много, чтобы пробегаться по массиву, менять в нем значение, ничего не сохранять в память и бежать дальше. У меня именно такой случай, поэтому и вопрос больше по ним (если не путаю их с итераторами, вот читаю про это сейчас).
Код запустить не удалось, там ошибка. В любом случае - спасибо! - что откликнулись).

[Игорь]

Timebird, я не знаком с датафреймами, поэтому смотрите на это не как на решение, а как на примерный вариант. Возможно они не позволяют поэлементное присваивание и надо как-то через методы.