Applying A Lambda Function To A Column Got Failed In Pandas

I don't know why the index method has inconsistent behavior while doing column-wise apply function. The data frame is: df = pd.DataFrame( [(1, 'Hello'), (2, 'World')]) df.columns=[

Solution 1:

When you index with 'B' you get a series. When you index with 1:2 or with ['B'], you get a DataFrame with one column. When you use apply on a series, your function is called on each element. When you use apply on a DataFrame, your function is called on each column.

So no, they aren't equivalent. When you have a Series you can use your function as you want. When you have a one-column DataFrame, you can't, because it gets passed the column as its argument, and the column is a Series that doesn't have an upper method.

You can see that they aren't the same because the results are different when you print them out. Yes, they're almost the same, but not the same. The first one has a column header, indicating that it's a DataFrame; the second has no column header but has the "Name" at the bottom, indicating it's a Series.

Solution 2:

As @BrenBarn mentioned, the difference is that in case of df.iloc[:, 1:2] you have DataFrame with one column, while in case of df.loc[:, 'B'] you have a Series. Just a little addition, to convert DataFrame with one column into series you can use pandas.squeeze() method:

>>> df.iloc[:, 1:2]B0Hello1World
>>> df.iloc[:, 1:2].squeeze()
0Hello1WorldName: B, dtype: object

and then you can use apply (you don't have to use lambda, BTW):

>>> df.iloc[:, 1:2].squeeze().apply(str.upper)
0HELLO1WORLDName: B, dtype: object

If you want to apply upper to DataFrame, you can use pandas.applymap():

>>>df.iloc[:, 1:2].applymap(str.upper)
       B
0  HELLO
1  WORLD