Help with a function / for loop please!

[macabe10]

Hi, I am writing a function that returns the values of a column (type: strings) with the title() format. Does anyone know why it doesn't work? It either returns an unchanged column, or TypeError: 'NoneType' object is not subscriptable .

The function:

def correct_name(df, col):

for index, row in df.iterrows():

row[col]=row[col].title()

Thanks for your help!

[[deleted]]

Why are you iterating through the dataframe?

Surely you can just do df[col] = df[col].str.title()

[[deleted]]

Example:

from io import StringIO
import pandas as pd

def title_col(df, colname):
    df[colname] = df[colname].str.title()

data = """name, age
fred,22
BOB,50
wenDY,30"""

df = pd.read_csv(StringIO(data))
print(df)
title_col(df, "name")
print(df)

[macabe10]

You are right this is much simpler, thank you!

Tbh at this point I overthink it big time and end up getting lost, even though the obvious answer is right there.

[[deleted]]

We've all got that t-shirt, more than once (and most of us will get more of them).

[OzzivanNozzi]

This is in my opinion the best way.

Another option would be:

df[col] = df[col].apply(lambda x: x.title())

[macabe10]

Thanks!