python pandas添加列_Python Pandas：基于其他列添加列

I'm new to pandas and pretty confused about it especially compared to lists and using list comprehensions.

I have a dataframe with 4 columns. I want to create a 5th column "c" based on 4th column "m". I can get the value for "c" by applying my function for each row in column "m".

If "m" was a list and using list comprehension it would be

c = [myfunction(x) for x in m]

How do I do apply this "logic" to a dataframe?

解决方案

You can assign - sample from doc:

import pandas as pd

import numpy as np

df = pd.DataFrame({'A': range(1, 11), 'B': np.random.randn(10)})

print df

A B

0 1 0.769028

1 2 -0.392471

2 3 0.153051

3 4 -0.379848

4 5 -0.665426

5 6 0.880684

6 7 1.126381

7 8 -0.559828

8 9 0.862935

9 10 -0.909402

df = df.assign(ln_A = lambda x: np.log(x.A))

print df

A B ln_A

0 1 0.769028 0.000000

1 2 -0.392471 0.693147

2 3 0.153051 1.098612

3 4 -0.379848 1.386294

4 5 -0.665426 1.609438

5 6 0.880684 1.791759

6 7 1.126381 1.945910

7 8 -0.559828 2.079442

8 9 0.862935 2.197225

9 10 -0.909402 2.302585

Or apply as Lu Qi commented.

Sometimes lambda function is helpful:

import pandas as pd

import numpy as np

df = pd.DataFrame({'A': range(1, 11), 'B': np.random.randn(10)})

df['ln_A'] = df['A'].apply(np.log)

df['round'] = df['B'].apply(lambda x: np.round(x, 2))

print df

A B ln_A round

0 1 -0.982828 0.000000 -0.98

1 2 2.306111 0.693147 2.31

2 3 0.967858 1.098612 0.97

3 4 -0.286280 1.386294 -0.29

4 5 -2.026937 1.609438 -2.03

5 6 0.061735 1.791759 0.06

6 7 -0.506620 1.945910 -0.51

7 8 -0.309438 2.079442 -0.31

8 9 -1.261842 2.197225 -1.26

9 10 1.079921 2.302585 1.08