python教程—使用panda查找最多两个或更多列-Python实用宝典

python教程—使用panda查找最多两个或更多列

我有一个列为a、B的dataframe。我需要创建一个列C,这样对于每条记录/行:C = max(a, B)。

I have a dataframe with columns A,B. I need to create a column C such that for every record / row:

C = max(A, B).

How should I go about doing this?

谢谢。

回答

You can get the maximum like this:

    >>> import pandas as pd >>> df = pd.DataFrame({"A": [1,2,3], "B": [-2, 8, 1]}) >>> df A B 0 1 -2 1 2 8 2 3 1 >>> df[["A", "B"]] A B 0 1 -2 1 2 8 2 3 1 >>> df[["A", "B"]].max(axis=1) 0 1 1 8 2 3

and so:

    >>> df["C"] = df[["A", "B"]].max(axis=1) >>> df A B C 0 1 -2 1 1 2 8 8 2 3 1 3

如果你知道“A”和“B”是唯一的两列,你甚至可以不受影响

    >>> df["C"] = df.max(axis=1)

And you could use .apply(max, axis=1) too, I guess.

​Python实用宝典 (pythondict.com)
不只是一个宝典
欢迎关注公众号:Python实用宝典

本文由 Python实用宝典 作者:Python实用宝典 发表,其版权均为 Python实用宝典 所有,文章内容系作者个人观点,不代表 Python实用宝典 对观点赞同或支持。如需转载,请注明文章来源。
1

发表评论